Skip to content
Snippets Groups Projects
Unverified Commit c74bb33a authored by fuzhongqian's avatar fuzhongqian Committed by Gitee
Browse files

update official/cv/resnet/golden_stick/quantization/simqat/train.py. 


Signed-off-by: default avatarfuzhongqian <fuzhongqian@huawei.com>
parent 8ced7bf4
Branches
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
official/cv/resnet/golden_stick/quantization/simqat/train.py
+ 2
− 2
View file @ c74bb33a
...@@ -259,8 +259,8 @@ def train_net(): ...@@ -259,8 +259,8 @@ def train_net():
loss_cb = LossCallBack(config.has_trained_epoch) loss_cb = LossCallBack(config.has_trained_epoch)
cb = [time_cb, loss_cb] cb = [time_cb, loss_cb]
if algo: if algo:
algo_cb = algo.callback() algo_cb = algo.callbacks()
cb.append(algo_cb) cb += algo_cb
ckpt_save_dir = set_save_ckpt_dir() ckpt_save_dir = set_save_ckpt_dir()
if config.save_checkpoint: if config.save_checkpoint:
ckpt_append_info = [{"epoch_num": config.has_trained_epoch, "step_num": config.has_trained_step}] ckpt_append_info = [{"epoch_num": config.has_trained_epoch, "step_num": config.has_trained_step}]
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment