val_multi.sh 358 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
6
#!/usr/bin/env bash
export HIP_VISIBLE_DEVICES=0,1,2,3 # 自行修改为训练的卡号和数量
export HSA_FORCE_FINE_GRAIN_PCIE=1
export USE_MIOPEN_BATCHNORM=1

# coco_path是训练数据集地址,数据是coco format
chenych's avatar
chenych committed
7

chenych's avatar
chenych committed
8
9
10
11
GPUS_PER_NODE=4 ./tools/run_dist_launch.sh 4 <config path> \
    --coco_path <coco path> \
    --resume <checkpoint path> \
    --eval