|
|
@@ -31,6 +31,7 @@ Torch_MNIST_Example_Model模型可从本项目的模型目录中引用,[预训 |
|
|
##### 1,gpu示例代码: |
|
|
##### 1,gpu示例代码: |
|
|
|
|
|
|
|
|
- 训练任务单卡示例请参考示例中[train.py](./train.py)的代码注释 |
|
|
- 训练任务单卡示例请参考示例中[train.py](./train.py)的代码注释 |
|
|
|
|
|
- 训练任务多卡示例请参考示例中[parallel_train.py](./parallel_train.py)的代码注释 |
|
|
- 推理任务示例请参考示例中[inference.py](./inference.py)的代码注释 |
|
|
- 推理任务示例请参考示例中[inference.py](./inference.py)的代码注释 |
|
|
- [model.py](./model.py)是运行项目所需的代码 |
|
|
- [model.py](./model.py)是运行项目所需的代码 |
|
|
|
|
|
|
|
|
|