val.sh

#!/bin/bash
echo "Testing start ..."
export HIP_VISIBLE_DEVICES=1
export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1

# resume: 待测试模型地址
# coco_path: 训练数据集地址, 数据是coco format
# dcu
# python main.py --batch_size 2 --no_aux_loss --eval --resume ./results/checkpoint.pth --coco_path /home/datasets/COCO2017
# ori
python main.py --batch_size 2 --no_aux_loss --eval --resume ./pretrained_model/detr-r50-e632da11.pth --coco_path /home/datasets/COCO2017