模型训练
我们使用imagepre中生成的镜像进行模型训练。
训练
我们可以先在本地进行训练,测试训练能否正常进行,再在UAI-Train平台上进行训练。
本地训练
我们需要把训练需要的数据文件映射到uhub.service.auto-ai.com.cn/uai_demo/crnn_cpu:latest。
sudo docker run -it -v /data/data:/data/data -v /data/output:/data/output uhub.service.auto-ai.com.cn/uai_demo/crnn_cpu:latest /bin/bash -c "/usr/bin/python /data/code/tools/train_shadownet.py"
我们可以在/data/output中查看相应的训练输出。
平台训练
上传训练数据
我们需要将/data/data下的文件上传到UFile或者UFS中,这里我们以UFile平台为例。 我们使用事先下载的UFile工具,进入filemgr-linux64.elf文件夹,通过UFile平台的数据上传命令将/data/data下的数据上传至UFile中,并以/crnn_poem/data/作为前缀,你可以自由地修改上传数据的前缀;
./filemgr-linux64 --action mput --bucket uai-demo --dir /data/data/tfrecords/ --prefix /crnn_poem/data/tfrecords/ ./filemgr-linux64 --action mput --bucket uai-demo --dir /data/data/char_dict/ --prefix /crnn_poem/data/char_dict/
填写数据输入路径
在UAI-Train平台上训练的具体操作步骤可以参考train
如果你的数据存放在UFile平台的/crnn_poem/data/下面,你的数据路径可以填写为
http://uai-demo.ufile.auto-ai.com.cn//crnn_poem/data/
执行训练命令
训练的执行命令为:
/data/code/tools/train_shadownet.py
我们可以在设置的输出路径中查看模型训练结果。