上一篇相关文章中实现了aster的demo脚本,继而我们实现aster算法的训练模型:
编写dockerfile:
修改train.py:

运行builders目录下的input_reader_builder.py, 用该脚本获得训练数据:
在data目录下生成

启动镜像:
docker run -i -t aster-test:v2 /bin/bash
运行train.py文件:
docker run --gpus all -v/data/ai_lls/qiao_q/aster-code/aster-master:/aster-code/aster-master/ -w/aster-code/aster-master/ asiter-train:v1 python train.py
之后在experiments/demo/log目录下找到相关checkpoints文件

之后会上传该算法相关的所有文件!

本文详细介绍如何通过编写Dockerfile及修改train.py脚本来实现Aster算法的训练模型,包括运行input_reader_builder.py获取训练数据,以及在指定目录下找到checkpoints文件的过程。
5695

被折叠的 条评论
为什么被折叠?



