diff --git a/gpu_mnist_example/README.md b/gpu_mnist_example/README.md index 4acbd34..3ed6fc8 100644 --- a/gpu_mnist_example/README.md +++ b/gpu_mnist_example/README.md @@ -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)是运行项目所需的代码