stable-diffusion-finetune/scripts/slurm/v1_edgeinpainting/launcher.sh

45 lines
1.6 KiB
Bash
Raw Normal View History

2022-08-03 00:53:58 +02:00
#!/bin/bash
# mpi version for node rank
H=`hostname`
THEID=`echo -e $HOSTNAMES | python3 -c "import sys;[sys.stdout.write(str(i)) for i,line in enumerate(next(sys.stdin).split(' ')) if line.strip() == '$H'.strip()]"`
export NODE_RANK=${THEID}
echo THEID=$THEID
echo "##########################################"
echo MASTER_ADDR=${MASTER_ADDR}
echo MASTER_PORT=${MASTER_PORT}
echo NODE_RANK=${NODE_RANK}
echo WORLD_SIZE=${WORLD_SIZE}
echo "##########################################"
# debug environment worked great so we stick with it
# no magic there, just a miniconda python=3.9, pytorch=1.12, cudatoolkit=11.3
# env with pip dependencies from stable diffusion's requirements.txt
eval "$(/fsx/stable-diffusion/debug/miniconda3/bin/conda shell.bash hook)"
#conda activate stable
# torch 1.11 to avoid bug in ckpt restoring
conda activate torch111
cd /fsx/stable-diffusion/stable-diffusion
CONFIG="/fsx/stable-diffusion/stable-diffusion/configs/stable-diffusion/inpainting/v1-edgeinpainting.yaml"
2022-08-04 00:22:07 +02:00
# start without masking
2022-08-06 12:42:32 +02:00
EXTRA="data.params.train.postprocess.params.mask_edges=false -f _extracat"
2022-08-04 00:22:07 +02:00
2022-08-03 00:53:58 +02:00
# resume and set new seed to reshuffle data
#EXTRA="--seed 543 --resume_from_checkpoint ..."
# reduce lr a bit
#EXTRA="${EXTRA} model.params.scheduler_config.params.f_max=[0.75]"
# custom logdir
#EXTRA="${EXTRA} --logdir rlogs"
# debugging
#EXTRA="${EXTRA} -d True lightning.callbacks.image_logger.params.batch_frequency=50"
# detect bad gpus early on
/bin/bash /fsx/stable-diffusion/stable-diffusion/scripts/test_gpu.sh
2022-08-04 00:22:07 +02:00
python main.py --base $CONFIG --gpus 0,1,2,3,4,5,6,7 -t --num_nodes ${WORLD_SIZE} --scale_lr False $EXTRA