pushd /data/flagperf/training/hygon/unet-paddle/extern 
/usr/local/bin/pip3 install -e .
popd
data_dir="data/cityscapes"
if [ ! -x "$data_dir" ]; then
  ln -s /data/unet/train/cityscapes $data_dir
fi
export HIP_VISIBLE_DEVICES=0,1,2,3
export HSA_FORCE_FINE_GRAIN_PCIE=1
export HIP_UPSAMPLE_OPTIMIZE=1

python -m paddle.distributed.launch tools/train.py --config configs/unet/unet_cityscapes_1024x512_160k.yml --do_eval --use_vdl --save_interval 176 --save_dir /data/unetouput --log_iters 1 $1 $2

