File size: 193 Bytes
111d9b6 |
1 2 3 4 5 6 |
#!/usr/bin/env bash
CONFIG=$1
PORT_=$2
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 --master_port=$PORT_ basicsr/train.py -opt $CONFIG --launcher pytorch
|