!10798 CPU data_sink_mode should be False

From: @caojian05
Reviewed-by: @wuxuejian,@oacjiewen
Signed-off-by: @wuxuejian
pull/10798/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 9b2cd73c3b

@ -170,7 +170,7 @@ def train():
ckpoint_cb = ModelCheckpoint(prefix=args.model, directory=args.train_dir, config=config_ck)
cbs.append(ckpoint_cb)
model.train(args.train_epochs, dataset, callbacks=cbs)
model.train(args.train_epochs, dataset, callbacks=cbs, dataset_sink_mode=(args.device_target != "CPU"))
if __name__ == '__main__':

Loading…
Cancel
Save