CONFIG="configs/body_2d_keypoint/rtmo/coco/rtmo-s_8xb32-600e_coco-640x640.py"
NUM_GPUS=4
# WEIGHT="checkpoint/epoch_xxx.pth"
WORK_DIR="checkpoint"
# sh tools/dist_train.sh $CONFIG $NUM_GPUS --resume $WEIGHT --work-dir $WORK_DIR
sh tools/dist_train.sh $CONFIG $NUM_GPUS --work-dir $WORK_DIR --amp
