5 lines
244 B
Bash
5 lines
244 B
Bash
|
gpu=2080Ti mWidth=448 mHeight=32
|
||
|
|
pth=/mnt/thsw2/DSP2/weights/ocr_en/english_g2
|
||
|
|
python pth2onnx.py --weights ${pth}.pth --mWidth ${mWidth} --mHeight ${mHeight}
|
||
|
|
mv ${pth}_${mWidth}X${mHeight}.engine $pth_${gpu}_fp16_${mWidth}X${mHeight}.engine
|