#!/usr/bin/env bash

CONFIG=$1
GPUS=$2

#  CUDA_VISIBLE_DEVICES=$GPUS \
PYTHONPATH=.:$PYTHONPATH OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node="$GPUS" \
  main.py train "$CONFIG" --backup_config --setup_output_dir --setup_random_seed