Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

CPU data_sink_mode should be False

tags/v1.2.0-rc1
CaoJian 5 years ago
parent
3e912d61d0
commit
ae2317a073
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      model_zoo/official/cv/deeplabv3/train.py

+ 1
- 1
model_zoo/official/cv/deeplabv3/train.py View File

@@ -170,7 +170,7 @@ def train():
ckpoint_cb = ModelCheckpoint(prefix=args.model, directory=args.train_dir, config=config_ck) ckpoint_cb = ModelCheckpoint(prefix=args.model, directory=args.train_dir, config=config_ck)
cbs.append(ckpoint_cb) 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__': if __name__ == '__main__':


Write Preview
Loading…
Cancel
Save