From 0b037348adf49cca923405e6afe1fb71ebf67786 Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Sat, 25 Feb 2023 16:37:10 -0500 Subject: [PATCH] update coco example cfg --- cfgs/train_coco.json | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,