Skip to content
Snippets Groups Projects
Commit 18f3fb5f authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!1681 fix some bugs in mobilenetv2 & vgg16

Merge pull request !1681 from zhaoting/master
parents 2ff80678 bffc4d3e
No related branches found
No related tags found
No related merge requests found
......@@ -37,7 +37,7 @@ platform: 'GPU'
device_id: 0
rank_id: 0
rank_size: 1
run_distribute: True
run_distribute: False
activation: "Softmax"
# Image classification trian. train_parse_args():return train_args
......
......@@ -221,7 +221,7 @@ def run_train():
# define callbacks
time_cb = TimeMonitor(data_size=batch_num)
loss_cb = LossMonitor(per_print_times=batch_num)
loss_cb = LossMonitor()
callbacks = [time_cb, loss_cb]
if config.rank_save_ckpt_flag:
ckpt_config = CheckpointConfig(save_checkpoint_steps=config.ckpt_interval * config.steps_per_epoch,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment