File size: 272 Bytes
4b7b610 | 1 2 3 4 5 6 7 8 9 | BASE_DATA_DIR="path/to/basedata"
BASE_CKPT_DIR="path/to/sd2_ckpt"
export CUDA_VISIBLE_DEVICES=3
python train_s1.py --config config/train_s1.yaml \
--base_data_dir $BASE_DATA_DIR \
--base_ckpt_dir $BASE_CKPT_DIR \
--output_dir log/stage1_bs8 \
--no_wandb \ |