diff --git a/main.py b/main.py index 1406decf3899d7a6f1d3d423a03dcb0278f4866e..aa86d51aa6a850498466a7a276616a9bf2eb96a6 100644 --- 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):