model_training.md
May 27, 2024 ยท View on GitHub
Train model
# single gpu
python tools/train.py $config
# multiple gpu
./tools/dist_train.sh $config num_gpu
Test model
# single gpu
python tools/test.py $config $checkpoint --eval mAP
# multiple gpu
./tools/dist_test.sh $config $checkpoint num_gpu --eval mAP
# ray-iou metric
./tools/dist_test.sh $config $checkpoint num_gpu --eval ray-iou
FPS for Panoptic-FlashOcc
# for single-frame
python tools/analysis_tools/benchmark.py config ckpt
python tools/analysis_tools/benchmark.py config ckpt --w_pano
# for multi-frame
python tools/analysis_tools/benchmark_sequential.py config ckpt
python tools/analysis_tools/benchmark_sequential.py config ckpt --w_pano