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

更新 'gpu_mnist_example/train_gcu.py'

liuzx-patch-1
liuzxtest02 2 years ago
parent
65d5abcc0c
commit
9624c366c5
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      gpu_mnist_example/train_gcu.py

+ 1
- 1
gpu_mnist_example/train_gcu.py View File

@@ -74,7 +74,7 @@ if __name__ == '__main__':
pretrain_model_path = openi_context.pretrain_model_path
output_path = openi_context.output_path

dataset_path_A = dataset_path + "/MnistDataset_torch"
dataset_path_A = dataset_path + "/MnistDataset"
pretrain_model_path_A = pretrain_model_path + "/MNIST_PytorchExample_GPU_test34_model_7f9j"

print("dataset_path:")


Write Preview
Loading…
Cancel
Save