Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!6663 fix multi-servers execution bug 

Merge pull request !6663 from guoqi/master
This commit is contained in:
mindspore-ci-bot 2020-09-23 14:23:45 +08:00 committed by Gitee
parent 9f84920f64 f228e647c4
commit 6ddf805807
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/resnet/train.py
View File

@ -188,4 +188,4 @@ if __name__ == '__main__':
if args_opt.net == "se-resnet50": if args_opt.net == "se-resnet50":
config.epoch_size = config.train_epoch_size config.epoch_size = config.train_epoch_size
model.train(config.epoch_size - config.pretrain_epoch_size, dataset, callbacks=cb, model.train(config.epoch_size - config.pretrain_epoch_size, dataset, callbacks=cb,
dataset_sink_mode=(not args_opt.parameter_server)) sink_size=dataset.get_dataset_size(), dataset_sink_mode=(not args_opt.parameter_server))