Browse Source

update gpu readme

pull/27/head
liuzx 1 year ago
parent
commit
ce2e65ec3b
1 changed files with 43 additions and 42 deletions
  1. +43
    -42
      gpu_mnist_example/README.md

+ 43
- 42
gpu_mnist_example/README.md View File

@@ -26,7 +26,49 @@ Torch_MNIST_Example_Model模型可从本项目的模型目录中引用,[预训
> ├── mnist_epoch1.pkl
>

## 二. 如何在云脑上获取代码路径,数据集路径,预训练模型路径,输出路径
## 二.GPU样例准备

##### 1,gpu示例代码:

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

##### 2,创建GPU调试任务

表1创建训练作业界面参数说明

| 参数名称 | 说明 |
| -------- | ----------------------------------------------------------------------------------- |
| 代码分支 | 选择仓库代码中要使用的代码分支,默认可选择master分支 |
| 镜像 | 镜像选择含有python和torch的镜像,推荐镜像: |
| 启动文件 | 启动文件选择代码目录下的启动脚本,在本示例中选择gpu_mnist_example/train.py |
| 数据集 | 数据集选择MnistDataset_torch.zip |
| 运行参数 | 选择增加运行参数可以向脚本中其他参数传值,如epoch_size,需要在代码里定义增加的超参数 |
| 资源规格 | 规格选择[GPU] |
| 模型 | 模型选择Torch_MNIST_Example_Model |

启动调试任务后,先执行prepare()进行数据准备;

进入对应的代码目录后,可在终端执行python train.py;

##### 3,创建GPU训练任务

表2 创建训练作业界面参数说明

| 参数名称 | 说明 |
| -------- | ----------------------------------------------------------------------------------- |
| 代码分支 | 选择仓库代码中要使用的代码分支,默认可选择master分支 |
| 镜像 | 镜像选择含有python和torch的镜像 |
| 启动文件 | 启动文件选择代码目录下的启动脚本,在本示例中选择gpu_mnist_example/train.py |
| 数据集 | 数据集选择MnistDataset_torch.zip |
| 运行参数 | 选择增加运行参数可以向脚本中其他参数传值,如epoch_size,需要在代码里定义增加的超参数 |
| 资源规格 | 规格选择[GPU] |
| 模型 | 模型选择Torch_MNIST_Example_Model |

启动训练任务后,训练结束会在任务的结果下载页提供输出结果下载

## 三. 如何在云脑上获取代码路径,数据集路径,预训练模型路径,输出路径

##### 1,准备数据

@@ -77,47 +119,6 @@ from c2net.context import prepare,upload_output
upload_output()
```

## 三.GPU样例准备

##### 1,gpu示例代码:

- 训练任务单卡示例请参考示例中[train.py](./train.py)的代码注释
- 推理任务示例请参考示例中[inference.py](./inference.py)的代码注释

##### 2,创建GPU调试任务

表1创建训练作业界面参数说明

| 参数名称 | 说明 |
| -------- | ----------------------------------------------------------------------------------- |
| 代码分支 | 选择仓库代码中要使用的代码分支,默认可选择master分支 |
| 镜像 | 镜像选择含有python和torch的镜像 |
| 启动文件 | 启动文件选择代码目录下的启动脚本,在本示例中选择gpu_mnist_example/train.py |
| 数据集 | 数据集选择MnistDataset_torch.zip |
| 运行参数 | 选择增加运行参数可以向脚本中其他参数传值,如epoch_size,需要在代码里定义增加的超参数 |
| 资源规格 | 规格选择[GPU] |
| 模型 | 模型选择Torch_MNIST_Example_Model |

启动调试任务后,先执行prepare()进行数据准备;

进入对应的代码目录后,可在终端执行python train.py;

##### 3,创建GPU训练任务

表2 创建训练作业界面参数说明

| 参数名称 | 说明 |
| -------- | ----------------------------------------------------------------------------------- |
| 代码分支 | 选择仓库代码中要使用的代码分支,默认可选择master分支 |
| 镜像 | 镜像选择含有python和torch的镜像 |
| 启动文件 | 启动文件选择代码目录下的启动脚本,在本示例中选择gpu_mnist_example/train.py |
| 数据集 | 数据集选择MnistDataset_torch.zip |
| 运行参数 | 选择增加运行参数可以向脚本中其他参数传值,如epoch_size,需要在代码里定义增加的超参数 |
| 资源规格 | 规格选择[GPU] |
| 模型 | 模型选择Torch_MNIST_Example_Model |

启动训练任务后,训练结束会在任务的结果下载页提供输出结果下载

## 四.GPU任务注意事项

##### 1, 使用超参数的方法


Loading…
Cancel
Save