diff --git a/_paramters.py b/_paramters.py index 17cac75..3abe2ea 100644 --- a/_paramters.py +++ b/_paramters.py @@ -30,12 +30,12 @@ main_arg_parser.add_argument("--data_stretch", type=strtobool, default=True, hel # Transformation Parameters main_arg_parser.add_argument("--data_loudness_ratio", type=float, default=0, help="") # 0.4 -main_arg_parser.add_argument("--data_shift_ratio", type=float, default=0.3, help="") # 0.3 +main_arg_parser.add_argument("--data_shift_ratio", type=float, default=0.3, help="") # 0.4 main_arg_parser.add_argument("--data_noise_ratio", type=float, default=0, help="") # 0.4 main_arg_parser.add_argument("--data_mask_ratio", type=float, default=0, help="") # 0.2 -main_arg_parser.add_argument("--data_speed_amount", type=float, default=0, help="") # 0.3 -main_arg_parser.add_argument("--data_speed_min", type=float, default=0.7, help="") # 0.7 -main_arg_parser.add_argument("--data_speed_max", type=float, default=1.7, help="") # 1.7 +main_arg_parser.add_argument("--data_speed_amount", type=float, default=0, help="") # 0.4 +main_arg_parser.add_argument("--data_speed_min", type=float, default=0, help="") # 0.7 +main_arg_parser.add_argument("--data_speed_max", type=float, default=0, help="") # 1.7 # Model Parameters main_arg_parser.add_argument("--model_type", type=str, default="RCC", help="") diff --git a/multi_run.py b/multi_run.py index 3f4191f..7c8b5e3 100644 --- a/multi_run.py +++ b/multi_run.py @@ -24,15 +24,18 @@ if __name__ == '__main__': arg_dict.update(main_seed=seed) for model in ['CC', 'BCMC', 'BCC', 'RCC']: arg_dict.update(model_type=model) - raw_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0, - data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.0, + raw_conf = dict(data_speed_amount=0.0, data_speed_min=0.0, data_speed_max=0.0, + data_mask_ratio=0.0, data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.0, data_stretch=False, train_epochs=401) - all_conf = dict(data_speed_factor=0.7, data_speed_ratio=0.4, data_mask_ratio=0.2, - data_noise_ratio=0.4, data_shift_ratio=0.4, data_loudness_ratio=0.4, + + all_conf = dict(data_speed_amount=0.4, data_speed_min=0.7, data_speed_max=1.7, + data_mask_ratio=0.2, data_noise_ratio=0.4, data_shift_ratio=0.4, data_loudness_ratio=0.4, data_stretch=True, train_epochs=101) + speed_conf = raw_conf.copy() speed_conf.update(data_speed_amount=0.4, data_speed_min=0.7, data_speed_max=1.7, data_stretch=True, train_epochs=101) + mask_conf = raw_conf.copy() mask_conf.update(data_mask_ratio=0.2, data_stretch=True, train_epochs=101)