From c2f8156c865018d62baf112b4c16d85021c75f46 Mon Sep 17 00:00:00 2001
From: anzhengqi <anzhengqi1@huawei.com>
Date: Thu, 25 Aug 2022 18:33:46 +0800
Subject: [PATCH] modify psenet scripts
---
official/cv/psenet/train.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/official/cv/psenet/train.py b/official/cv/psenet/train.py
index ab8ce5217..cd2b6c981 100644
--- a/official/cv/psenet/train.py
+++ b/official/cv/psenet/train.py
@@ -118,9 +118,9 @@ def train():
net = TrainOneStepCell(net, opt)
time_cb = TimeMonitor(data_size=step_size)
- loss_cb = LossCallBack(per_print_times=10)
+ loss_cb = LossCallBack(per_print_times=step_size)
# set and apply parameters of check point config.TRAIN_MODEL_SAVE_PATH
- ckpoint_cf = CheckpointConfig(save_checkpoint_steps=1875, keep_checkpoint_max=3)
+ ckpoint_cf = CheckpointConfig(save_checkpoint_steps=20*step_size, keep_checkpoint_max=10)
ckpoint_cb = ModelCheckpoint(prefix="PSENet",
config=ckpoint_cf,
directory="{}/ckpt_{}".format(config.TRAIN_MODEL_SAVE_PATH,
--
GitLab