From 8125c217b721662f2afc4083c0e09dfbb83af574 Mon Sep 17 00:00:00 2001 From: liuzx Date: Wed, 30 Jul 2025 11:03:40 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu=5Fmnist=5Fexample/REA?= =?UTF-8?q?DME.md'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gpu_mnist_example/README.md | 1 + 1 file changed, 1 insertion(+) 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)是运行项目所需的代码