Optical-Generative-models/iterative_train.sh at main · TanGeeGo/Optical-Generative-models · 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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#!/bin/bash

export CUDA_VISIBLE_DEVICES=0

# activate you conda env if not
# conda activate your_env_name

# run the training
## single gpu start
python main.py \
    --task iterative_optical \
    --num_gpu 1 \
    --data_path ./data/your_dataset_path \
    --output_dir ./logs/exp \
    --sample_size 32 \
    --in_channels 3 \
    --out_channels 3 \
    --num_epochs 100 \
    --train_batch_size 100 \
    --eval_batch_size 64 \
    --learning_rate_digital 1e-4 \
    --learning_rate_optical 5e-3 \
    --ddpm_num_steps 1000 \
    --ddpm_beta_schedule linear \
    --mixed_precision no \
    --seed 96 \
    --save_image_epochs 10 \
    --save_model_epochs 50 \
    --lr_warmup_steps 100 \
    --gradient_accumulation_steps 1 \
    --num_classes 0 \
    --c 299792458.0 \
    --ridx_air 1.0 \
    --object_layer_dist 0.05 \
    --layer_layer_dist 0.01 \
    --layer_sensor_dist 0.05 \
    --num_layer 5 \
    --total_num 800 \
    --obj_num 320 \
    --layer_neuron_num 400 \
    --dxdy 8e-6 \
    --layer_init_method zero \
    --amp_modulation False \
    --time_embedding_type_d positional \
    --wavelength_mtcl 4.5e-7 5.2e-7 6.38e-7 \
    --ridx_layer_mtcl 1.0 1.0 1.0 \
    --attenu_factor_mtcl 0.0 0.0 0.0 \
    --apply_scale_itrt True \
    --scale_type_itrt static_mean \
    --prediction_type_o sample \
    --beta_start_itrt 0.001 \
    --beta_end_itrt 0.010 \

## multi gpu start
# accelerate launch --multi_gpu --num_processes 4 main.py \
#     --task iterative_optical \
#     --num_gpu 4 \
#     --data_path ./data/your_dataset_path \
#     --output_dir ./logs/exp \
#     --sample_size 32 \
#     --in_channels 3 \
#     --out_channels 3 \
#     --num_epochs 100 \
#     --train_batch_size 400 \
#     --eval_batch_size 64 \
#     --learning_rate_digital 1e-4 \
#     --learning_rate_optical 5e-3 \
#     --ddpm_num_steps 1000 \
#     --ddpm_beta_schedule linear \
#     --mixed_precision no \
#     --seed 96 \
#     --save_image_epochs 10 \
#     --save_model_epochs 50 \
#     --lr_warmup_steps 100 \
#     --gradient_accumulation_steps 1 \
#     --num_classes 0 \
#     --c 299792458.0 \
#     --ridx_air 1.0 \
#     --object_layer_dist 0.05 \
#     --layer_layer_dist 0.01 \
#     --layer_sensor_dist 0.05 \
#     --num_layer 5 \
#     --total_num 800 \
#     --obj_num 320 \
#     --layer_neuron_num 400 \
#     --dxdy 8e-6 \
#     --layer_init_method zero \
#     --amp_modulation False \
#     --time_embedding_type_d positional \
#     --wavelength_mtcl 4.5e-7 5.2e-7 6.38e-7 \
#     --ridx_layer_mtcl 1.0 1.0 1.0 \
#     --attenu_factor_mtcl 0.0 0.0 0.0 \
#     --apply_scale_itrt True \
#     --scale_type_itrt static_mean \
#     --prediction_type_o sample \
#     --beta_start_itrt 0.001 \
#     --beta_end_itrt 0.010 \