steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

_paramters.py and multi_run.py fixed accordingly

Browse Source
This commit is contained in:
Si11ium 2020-05-21 14:49:07 +02:00
parent e4abbc045c
commit e21143045a
2 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
_paramters.py
View File

@ -30,12 +30,12 @@ main_arg_parser.add_argument("--data_stretch", type=strtobool, default=True, hel
# Transformation Parameters # Transformation Parameters
main_arg_parser.add_argument("--data_loudness_ratio", type=float, default=0, help="") # 0.4 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_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_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_amount", type=float, default=0, help="") # 0.4
main_arg_parser.add_argument("--data_speed_min", type=float, default=0.7, help="") # 0.7 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=1.7, help="") # 1.7 main_arg_parser.add_argument("--data_speed_max", type=float, default=0, help="") # 1.7
# Model Parameters # Model Parameters
main_arg_parser.add_argument("--model_type", type=str, default="RCC", help="") main_arg_parser.add_argument("--model_type", type=str, default="RCC", help="")

11
multi_run.py
View File

@ -24,15 +24,18 @@ if __name__ == '__main__':
arg_dict.update(main_seed=seed) arg_dict.update(main_seed=seed)
for model in ['CC', 'BCMC', 'BCC', 'RCC']: for model in ['CC', 'BCMC', 'BCC', 'RCC']:
arg_dict.update(model_type=model) arg_dict.update(model_type=model)
raw_conf = dict(data_speed_factor=0.0, data_speed_ratio=0.0, data_mask_ratio=0.0, raw_conf = dict(data_speed_amount=0.0, data_speed_min=0.0, data_speed_max=0.0,
data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=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) 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) data_stretch=True, train_epochs=101)
speed_conf = raw_conf.copy() speed_conf = raw_conf.copy()
speed_conf.update(data_speed_amount=0.4, data_speed_min=0.7, data_speed_max=1.7, speed_conf.update(data_speed_amount=0.4, data_speed_min=0.7, data_speed_max=1.7,
data_stretch=True, train_epochs=101) data_stretch=True, train_epochs=101)
mask_conf = raw_conf.copy() mask_conf = raw_conf.copy()
mask_conf.update(data_mask_ratio=0.2, data_stretch=True, train_epochs=101) mask_conf.update(data_mask_ratio=0.2, data_stretch=True, train_epochs=101)