Browse Source

更新 'npu_mnist_example/train_npu.py'

liuzx-patch-1
liuzxtest02 2 years ago
parent
commit
079e71ecf7
1 changed files with 1 additions and 4 deletions
  1. +1
    -4
      npu_mnist_example/train_npu.py

+ 1
- 4
npu_mnist_example/train_npu.py View File

@@ -53,10 +53,7 @@ if __name__ == "__main__":
args, unknown = parser.parse_known_args()
data_dir = ''
pretrain_dir = ''
train_dir = ''

#回传结果到openi
upload_openi()
train_dir = ''
device_num = int(os.getenv('RANK_SIZE'))
#使用单卡时


Loading…
Cancel
Save