pushd /data/flagperf/training/hygon/hrnet-pytorch/extern 
/usr/local/bin/pip3 install -e .
popd
data_dir="data/coco"
if [ ! -x "$data_dir" ]; then
  ln -s /data/hrnet/train/COCO2017 data/coco
fi
export HIP_VISIBLE_DEVICES=0,1,2,3
export HSA_FORCE_FINE_GRAIN_PCIE=1
export HIP_UPSAMPLE_OPTIMIZE=1
#./tools/dist_train.sh configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512.py 4 --cfg-options model.pretrained=/data/hrnet/train/hrnet_w32_coco_512x512-bcb8c247_20200816.pth evaluation.interval=50 optimizer.lr=0.0005 lr_config.warmup_ratio=0.0002 --work-dir /data/hrnetout $1 $2
./tools/dist_train.sh configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512.py 4 --cfg-options evaluation.interval=50 --work-dir /data/hrnetout $1 $2
