diff --git a/cfgs/train_coco.json b/cfgs/train_coco.json index a8232dd..016f305 100644 --- a/cfgs/train_coco.json +++ b/cfgs/train_coco.json @@ -19,7 +19,7 @@ "lr_decay_steps": 0, "lr_scheduler": "constant", "lr_warmup_steps": null, - "max_epochs": 30, + "max_epochs": 10, "notebook": false, "optimizer_config": "optimizer.json", "project_name": "coco_test", @@ -30,7 +30,7 @@ "sample_steps": 300, "save_ckpt_dir": null, "save_ckpts_from_n_epochs": 0, - "save_every_n_epochs": 20, + "save_every_n_epochs": 1, "save_optimizer": false, "scale_lr": false, "seed": 555,