wangxiaoman/modelscope-cuda repository overview

ubuntu20.04-cuda11.8.0-py38-torch2.0.1-tf2.13.0-1.9.5

data_path(cv数据):/train_data/

训练数据格式：lmdb

数据集合：mnist，hwdb，modelscope数据集ICDAR13_HCTR_Dataset

模型output：/tmp/tmp8y7borva/output

容器启动命令：docker run -d --shm-size=10g -it --entrypoint /bin/bash -v /data/home/wangxiaoman/train_data:/train_data wangxiaoman/modelscope-cuda:v1

训练代码：/app/table_ocr/train_model/train_model.py