HolisticDiffuScene/run_st3d_room_layout_train.sh at develop · fangchuan/HolisticDiffuScene · 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
# bedroom parametrization
MODEL_FLAGS="--layout_channels 32 --layout_size 23 --num_channels 128 --num_res_blocks 3 --b_learn_sigma True  --b_class_cond False --b_text_cond True --use_input_encoding True"
MAX_STEPS=400000
# living room parametrization
# MODEL_FLAGS="--layout_channels 34 --layout_size 44 --num_channels 128 --num_res_blocks 3 --b_learn_sigma True --b_class_cond False --b_text_cond True --use_input_encoding True"


DIFFUSION_FLAGS="--diffusion_steps 4000 --noise_schedule cosine"
TRAIN_FLAGS="--lr 1e-4 --batch_size 64 --schedule_sampler loss-second-moment --use_3d_iou False "
NUM_GPUS=2

mpiexec -n $NUM_GPUS python scripts/st3d_room_layout_train.py --data_dir /mnt/nas_3dv/hdd1/datasets/Structured3d/preprocessed/new_text2pano/train/bedroom/ \
 $MODEL_FLAGS \
 $DIFFUSION_FLAGS \
 $TRAIN_FLAGS \
 --lr_anneal_steps $MAX_STEPS \
 --log_dir log/ST3D_bedroom/textcondition

# nsys profile -w true -t cuda,nvtx,osrt,cudnn,cublas -s cpu  --capture-range=cudaProfilerApi --capture-range-end=stop --cudabacktrace=true -x true -o my_profile_4 \
# mpiexec -n 1 python scripts/st3d_room_layout_train.py --data_dir /mnt/nas_3dv/hdd1/datasets/Structured3d/preprocessed/0919_new_livingroom/train/livingroom/ \
#  $MODEL_FLAGS \
#  $DIFFUSION_FLAGS \
#  $TRAIN_FLAGS  --lr_anneal_steps 101\
#  --log_dir log/ST3D_livingroom_textcondition_openai-2023-10-12-09-00-00-00