修改训练模型保存设置

main.py

@@ -34,8 +34,8 @@ 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=1875, keep_checkpoint_max=10)
-ckpoint = ModelCheckpoint(prefix="checkpoint_lenet", config=config_ck)
+config_ck = CheckpointConfig(save_checkpoint_steps=1, keep_checkpoint_max=10)
+ckpoint = ModelCheckpoint(prefix="deeplab.ckpt", config=config_ck)
 
 def cal_hist(a, b, n):
     k = (a >= 0) & (a < n)