10 lines
456 B
Bash
10 lines
456 B
Bash
|
#动态尺寸的stdc转trt没有成功，因为里有用到了全局池化，池化的过程中kener_size不固定
|
|||
|
|
gpu=2080Ti
|
|||
|
|
name=crackMeasurement
|
|||
|
|
nclass=2
|
|||
|
|
mWidth=0
|
|||
|
|
mHeight=0
|
|||
|
|
python toTrt.py --weights /mnt/thsw2/DSP2/weights/${name}/stdc_360X640.pth --nclass ${nclass} --mWidth ${mWidth} --mHeight ${mHeight}
|
|||
|
|
|
|||
|
|
mv /mnt/thsw2/DSP2/weights/${name}/stdc_${mWidth}X${mHeight}.engine /mnt/thsw2/DSP2/weights/${name}/stdc_${mHeight}X${mWidth}_${gpu}_fp16.engine
|