AIlib2/crowdUtils/run.sh

5 lines
195 B
Bash

gpu=2080Ti
url=/mnt/thsw2/DSP2/weights/crowdCounting/weights
python toTrt.py --weights ${url}/crowdCounting.pth
mv ${url}/crowdCounting_dynamic.engine ${url}/crowdCounting_${gpu}_dynamic.engine