diff --git a/_paramters.py b/_paramters.py index b197635..f707231 100644 --- a/_paramters.py +++ b/_paramters.py @@ -31,9 +31,9 @@ main_arg_parser.add_argument("--data_mixup", type=strtobool, default=False, help # Transformation Parameters main_arg_parser.add_argument("--data_loudness_ratio", type=float, default=0.4, help="") -main_arg_parser.add_argument("--data_shift_ratio", type=float, default=0, help="") -main_arg_parser.add_argument("--data_noise_ratio", type=float, default=0, help="") -main_arg_parser.add_argument("--data_mask_ratio", type=float, default=0, help="") +main_arg_parser.add_argument("--data_shift_ratio", type=float, default=0.3, help="") +main_arg_parser.add_argument("--data_noise_ratio", type=float, default=0.4, help="") +main_arg_parser.add_argument("--data_mask_ratio", type=float, default=0.2, help="") main_arg_parser.add_argument("--data_speed_ratio", type=float, default=0.3, help="") main_arg_parser.add_argument("--data_speed_factor", type=float, default=0.7, help="") @@ -41,9 +41,9 @@ main_arg_parser.add_argument("--data_speed_factor", type=float, default=0.7, hel main_arg_parser.add_argument("--train_outpath", type=str, default="output", help="") main_arg_parser.add_argument("--train_version", type=strtobool, required=False, help="") # FIXME: Stochastic weight Avaraging is not good, maybe its my implementation? -main_arg_parser.add_argument("--train_sto_weight_avg", type=strtobool, default=True, help="") +main_arg_parser.add_argument("--train_sto_weight_avg", type=strtobool, default=False, help="") main_arg_parser.add_argument("--train_opt_reset_interval", type=int, default=0, help="") -main_arg_parser.add_argument("--train_epochs", type=int, default=101, help="") +main_arg_parser.add_argument("--train_epochs", type=int, default=51, help="") main_arg_parser.add_argument("--train_batch_size", type=int, default=300, help="") main_arg_parser.add_argument("--train_lr", type=float, default=1e-4, help="") main_arg_parser.add_argument("--train_num_sanity_val_steps", type=int, default=0, help="") diff --git a/main.py b/main.py index d429e69..225fb8f 100644 --- a/main.py +++ b/main.py @@ -50,7 +50,7 @@ def run_lightning_loop(config_obj): show_progress_bar=True, weights_save_path=logger.log_dir, gpus=[0] if torch.cuda.is_available() else None, - check_val_every_n_epoch=10, + check_val_every_n_epoch=5, # num_sanity_val_steps=config_obj.train.num_sanity_val_steps, # row_log_interval=(model.n_train_batches * 0.1), # TODO: Better Value / Setting # log_save_interval=(model.n_train_batches * 0.2), # TODO: Better Value / Setting