val.sh 509 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
#!/bin/bash
echo "Testing start ..."
export HIP_VISIBLE_DEVICES=1
export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1
Rayyyyy's avatar
Rayyyyy committed
6
export MIOPEN_FIND_MODE=4
chenych's avatar
chenych committed
7
8
9

# resume: 待测试模型地址
# coco_path: 训练数据集地址, 数据是coco format
chenych's avatar
chenych committed
10
python main.py --batch_size 2 --no_aux_loss --eval --resume /path/of/model --coco_path /path/of/coco_data
chenych's avatar
chenych committed
11
# python main.py --batch_size 2 --no_aux_loss --eval --resume https://dl.fbaipublicfiles.com/detr/detr-r50-e632da11.pth --coco_path /home/datasets/COCO2017