修改训练模型保存设置

9afcdf36 · hexiangdong2019 · 1409f38e · 9afcdf36
Commit 9afcdf36 authored 3 years ago by hexiangdong2019
--- a/main.py
+++ b/main.py
@@ -34,7 +34,7 @@ lr_iter = cosine_lr(3e-5, 20, 500)
 net_loss = SoftmaxCrossEntropyLoss(6, 255)
 net_opt = nn.Adam(net.trainable_params(), learning_rate=lr_iter)

-config_ck = CheckpointConfig(save_checkpoint_steps=1, keep_checkpoint_max=10)
+config_ck = CheckpointConfig(save_checkpoint_steps=30, keep_checkpoint_max=1)
 ckpoint = ModelCheckpoint(prefix="deeplab.ckpt", config=config_ck)

 def cal_hist(a, b, n):