9 lines
258 B
Bash
9 lines
258 B
Bash
#!/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
|