diff --git a/_paramters.py b/_paramters.py index 97bc52d..6957169 100644 --- a/_paramters.py +++ b/_paramters.py @@ -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=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_lr", type=float, default=1e-4, help="") main_arg_parser.add_argument("--train_num_sanity_val_steps", type=int, default=0, help="")