Files
fcb_photo_review/services/paddle_services/doc_dewarp/train.sh
2024-09-24 17:10:56 +08:00

11 lines
217 B
Bash

export OPENCV_IO_ENABLE_OPENEXR=1
export FLAGS_logtostderr=0
export CUDA_VISIBLE_DEVICES=0
python train.py --img-size 288 \
--name "DocTr++" \
--batch-size 12 \
--lr 1e-4 \
--exist-ok \
--use-vdl