Residual Model

This commit is contained in:
Si11ium 2020-05-13 14:56:00 +02:00
parent 52d83a5006
commit c4676ea332

View File

@ -30,7 +30,7 @@ main_arg_parser.add_argument("--data_n_fft", type=int, default=512, help="")
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, help="")
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="")
@ -43,7 +43,7 @@ main_arg_parser.add_argument("--train_version", type=strtobool, required=False,
# 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_opt_reset_interval", type=int, default=0, help="")
main_arg_parser.add_argument("--train_epochs", type=int, default=31, help="")
main_arg_parser.add_argument("--train_epochs", type=int, default=101, 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="")