Test Dataset Multiplication by Timeshift

This commit is contained in:
Si11ium 2020-05-12 18:48:38 +02:00
parent dce799a52b
commit 52d83a5006

View File

@ -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? # 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=True, 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=30, help="") main_arg_parser.add_argument("--train_epochs", type=int, default=31, 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="")