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

更新 'npu_mnist_example/train_npu.py'

liuzx-patch-1
liuzxtest02 2 years ago
parent
fe8d22e849
commit
079e71ecf7
1 changed files with 1 additions and 4 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  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() args, unknown = parser.parse_known_args()
data_dir = '' data_dir = ''
pretrain_dir = '' pretrain_dir = ''
train_dir = ''

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


Write Preview
Loading…
Cancel
Save