Browse Source

更新 'gpu_mnist_example/README.md'

liuzx-patch-18
liuzx 8 months ago
parent
commit
8125c217b7
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      gpu_mnist_example/README.md

+ 1
- 0
gpu_mnist_example/README.md View File

@@ -31,6 +31,7 @@ Torch_MNIST_Example_Model模型可从本项目的模型目录中引用,[预训
##### 1,gpu示例代码:

- 训练任务单卡示例请参考示例中[train.py](./train.py)的代码注释
- 训练任务多卡示例请参考示例中[parallel_train.py](./parallel_train.py)的代码注释
- 推理任务示例请参考示例中[inference.py](./inference.py)的代码注释
- [model.py](./model.py)是运行项目所需的代码



Loading…
Cancel
Save