shapewords_paper_code/train_on_sample_data.sh at main · lodurality/shapewords_paper_code · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/env bash

PORT=29814
MODEL_NAME="stabilityai/stable-diffusion-2-1-base"
data_root=sample_data
categories_path=$data_root/categories.json
textured_images_dir=$data_root/controlnet_images_offset_all/
prompts_path=$data_root/foreground.txt
pb_emb_path=$data_root/shapenet_pointbert_tokens/
output_dir=sample_outputs
gg_depth=6
gg_heads=8
gg_drop_path_rate=0.1
exp_name=token_guidance_d_${gg_depth}_h_${gg_heads}_drop_${gg_drop_path_rate}_loss_weight_by_t/
split_path=stats/train_val.txt

accelerate launch --main_process_port $PORT train_geometry_guidance.py \
  --pretrained_model_name_or_path=$MODEL_NAME \
  --categories_path=$categories_path \
  --textured_images_dir=$textured_images_dir \
  --pb_emb_path=$pb_emb_path \
  --prompts_path=$prompts_path \
  --train_batch_size=24 \
  --gradient_accumulation_steps=1 \
  --num_train_epochs=100 \
  --learning_rate=5.0e-05 \
  --lr_warmup_steps=1000 \
  --output_dir="test_guidance" \
  --max_images=10000000 \
  --checkpointing_steps=1000 \
  --dataloader_num_workers=4 \
  --output_dir=$output_dir/$exp_name \
  --gg_depth=$gg_depth \
  --gg_heads=$gg_heads \
  --gg_drop_path_rate=$gg_drop_path_rate \
  --use_pb_tokens \
  --split_path=$split_path \
  --lr_scheduler=constant_with_warmup \
  --weight_loss_by_t \
  --loss_weight_m=500 \
  --loss_weight_s=250 \
  --use_transform \
  --min_crop_scale=0.8