diff --git a/train.py b/train.py index 223b2d9..7b73515 100644 --- a/train.py +++ b/train.py @@ -206,7 +206,7 @@ if __name__ == "__main__": parser.add_argument("--test_iterations", nargs="+", type=int, default=[7_000, 30_000]) parser.add_argument("--save_iterations", nargs="+", type=int, default=[7_000, 30_000]) parser.add_argument("--quiet", action="store_true") - parser.add_argument("--checkpoint_iterations", nargs="+", type=int, default=[]) + parser.add_argument("--checkpoint_iterations", nargs="+", type=int, default=[7_000, 30_000]) parser.add_argument("--start_checkpoint", type=str, default = None) parser.add_argument('--cuda_blocking', action='store_true', default=True) args = parser.parse_args(sys.argv[1:])