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

Residual Model

Browse Source
This commit is contained in:
Si11ium
2020-05-13 19:49:25 +02:00
parent c4676ea332
commit f3335d508a
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
_paramters.py
+5 -5
View File
@@ -31,9 +31,9 @@ main_arg_parser.add_argument("--data_mixup", type=strtobool, default=False, help
# Transformation Parameters # Transformation Parameters
main_arg_parser.add_argument("--data_loudness_ratio", type=float, default=0.4, 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_shift_ratio", type=float, default=0.3, help="")
main_arg_parser.add_argument("--data_noise_ratio", type=float, default=0, 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, 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_ratio", type=float, default=0.3, help="")
main_arg_parser.add_argument("--data_speed_factor", type=float, default=0.7, 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_outpath", type=str, default="output", help="")
main_arg_parser.add_argument("--train_version", type=strtobool, required=False, help="") main_arg_parser.add_argument("--train_version", type=strtobool, required=False, help="")
# FIXME: Stochastic weight Avaraging is not good, maybe its my implementation? # 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_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_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_lr", type=float, default=1e-4, help="")
main_arg_parser.add_argument("--train_num_sanity_val_steps", type=int, default=0, help="") main_arg_parser.add_argument("--train_num_sanity_val_steps", type=int, default=0, help="")
main.py
+1 -1
View File
@@ -50,7 +50,7 @@ def run_lightning_loop(config_obj):
show_progress_bar=True, show_progress_bar=True,
weights_save_path=logger.log_dir, weights_save_path=logger.log_dir,
gpus=[0] if torch.cuda.is_available() else None, 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, # num_sanity_val_steps=config_obj.train.num_sanity_val_steps,
# row_log_interval=(model.n_train_batches * 0.1), # TODO: Better Value / Setting # 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 # log_save_interval=(model.n_train_batches * 0.2), # TODO: Better Value / Setting