base code for pytorch distributed, add cyclegan

This commit is contained in:
2020-08-07 09:48:09 +08:00
commit f7843de45d
32 changed files with 1444 additions and 0 deletions

8
run.sh Normal file
View File

@@ -0,0 +1,8 @@
#!/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