From f228e647c4477a3d3273bca97c65cb674fa8c021 Mon Sep 17 00:00:00 2001 From: guoqi Date: Mon, 21 Sep 2020 19:46:52 +0800 Subject: [PATCH] fix multi-server execution bug by adding sink size parameter. --- model_zoo/official/cv/resnet/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/resnet/train.py b/model_zoo/official/cv/resnet/train.py index 3c29241214..a70b3b614b 100755 --- a/model_zoo/official/cv/resnet/train.py +++ b/model_zoo/official/cv/resnet/train.py @@ -188,4 +188,4 @@ if __name__ == '__main__': if args_opt.net == "se-resnet50": config.epoch_size = config.train_epoch_size 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))