HolisticDiffuScene/run_3dfront_bedroom_layout_train_sample.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
25
26
27
28
29
30
31
MODEL_FLAGS="--layout_channels 30 --layout_size 13 --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
ROOM_TYPE="bedroom"
CONFIG_FILE="/mnt/nas_3dv/hdd1/fangchuan/HolisticDiffuScene/config/3dfront_bedroom_config.yaml"
MAX_STEPS=400000

mpiexec -n $NUM_GPUS python scripts/threed_front_room_layout_train2.py \
    $MODEL_FLAGS \
    $DIFFUSION_FLAGS \
    $TRAIN_FLAGS \
    --room_type $ROOM_TYPE  \
    --config_file $CONFIG_FILE \
    --lr_anneal_steps $MAX_STEPS \
    --log_dir "log/3dfront_bedroom/textconditional-0122/"


MODEL_FLAGS="--layout_channels 30 --layout_size 13 --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  --timestep_respacing ddim200 --use_ddim True"
CONFIG_FILE="/mnt/nas_3dv/hdd1/fangchuan/HolisticDiffuScene/config/3dfront_bedroom_config.yaml"

python scripts/threed_front_room_layout_sample2.py \
    --model_path log/3dfront_bedroom/textconditional-0122/ema_0.9999_400000.pt \
     $MODEL_FLAGS \
     $DIFFUSION_FLAGS \
     --num_samples 1000 \
      --room_type $ROOM_TYPE \
      --path_to_pickled_3d_futute_models /mnt/nas_3dv/hdd1/datasets/3D_FRONT_FUTURE/threed_future_model_bedroom.pkl \
      --config_file $CONFIG_FILE \
      --log_dir "log/3dfront_bedroom/textconditional-0122/sample_results"