- 训练模型
目标任务:在华为npu上复现论文avatarGAN,并保证精度。
用的工具感觉都好陌生,入门教程也有好多对不上的地方,做起来感觉充满了阻力。之前遇到搞不定的都喜欢找别人帮我弄,这次想强迫自己做一遍,练下心态。
起始状态:npu代码和数据集已准备好(上传到自己的OBS目录下),ModalArts插件已安装。
训练模型- 在PyCharm工具栏中,选择“ModelArts > Edit Training Job Configuration”。
- 在弹出的对话框中,每个参数的意义如下。
- “Job Name”:自动生成,首次提交训练作业时,该名称也可以自己指定。
- “AI Engine”:选择训练的框架,以及版本。 本任务基于昇腾AI处理器使用Tensorflow训练框架执行训练。
- “Algorithm Source”:选择“Frequently-used”,代表常用框架。
- “Boot File Path”:选择本地的训练脚本“main.py”。
- “Code Directory”:选择启动脚本所在“src”目录。(填完“Boot File Path”会自动填上)
- “OBS Path”:填写输出路径,用于存储训练输出模型和日志文件。
- “Data Path in OBS”:填写数据上传的OBS目录。此处需完整OBS路径,需包含OBS桶名称。
- “Specifications”:选择GPU规格。
- “Running Parameters”:是训练脚本所需要的输入参数。
填写完成后,单击“Apply and Run”提交训练作业到云上ModelArts。



