-
Notifications
You must be signed in to change notification settings - Fork 34
/
Copy pathtrain_ldm.sh
82 lines (67 loc) · 3.68 KB
/
train_ldm.sh
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
#!/bin/bash\
### Train the Latent Diffusion Model ###
# --data_aug is optional
# max_face 30, max_edge 20 for deepcad
# max_face 50, max_edge 30 for abc/furniture
# --surfvae refer to the surface vae weights
# --edgevae refer to the edge vae weights
### Training DeepCAD Latent Diffusion Model ###
python ldm.py --data data_process/deepcad_parsed \
--list data_process/deepcad_data_split_6bit.pkl --option surfpos --gpu 0 1 \
--env deepcad_ldm_surfpos --train_nepoch 3000 --test_nepoch 200 --save_nepoch 200 \
--max_face 30 --max_edge 20
python ldm.py --data data_process/deepcad_parsed \
--list data_process/deepcad_data_split_6bit.pkl --option surfz \
--surfvae proj_log/deepcad_vae_surf.pt --gpu 0 1 \
--env deepcad_ldm_surfz --train_nepoch 3000 --batch_size 256 \
--max_face 30 --max_edge 20
python ldm.py --data data_process/deepcad_parsed \
--list data_process/deepcad_data_split_6bit.pkl --option edgepos \
--surfvae proj_log/deepcad_vae_surf.pt --gpu 0 1 \
--env deepcad_ldm_edgepos --train_nepoch 1000 --batch_size 128 \
--max_face 30 --max_edge 20
python ldm.py --data data_process/deepcad_parsed \
--list data_process/deepcad_data_split_6bit.pkl --option edgez \
--surfvae proj_log/deepcad_vae_surf.pt --edgevae proj_log/deepcad_vae_edge.pt --gpu 0 1 \
--env deepcad_ldm_edgez --train_nepoch 1000 --batch_size 128 \
--max_face 30 --max_edge 20
### Training ABC Latent Diffusion Model ###
python ldm.py --data data_process/abc_parsed \
--list data_process/abc_data_split_6bit.pkl --option surfpos --gpu 0 1 \
--env abc_ldm_surfpos --train_nepoch 1000 --test_nepoch 200 --save_nepoch 200 \
--max_face 50 --max_edge 30
python ldm.py --data data_process/abc_parsed \
--list data_process/abc_data_split_6bit.pkl --option surfz \
--surfvae proj_log/abc_vae_surf.pt --gpu 0 1 \
--env abc_ldm_surfz --train_nepoch 1000 --batch_size 256 \
--max_face 50 --max_edge 30
python ldm.py --data data_process/abc_parsed \
--list data_process/abc_data_split_6bit.pkl --option edgepos \
--surfvae proj_log/abc_vae_surf.pt --gpu 0 1 \
--env abc_ldm_edgepos --train_nepoch 300 --batch_size 64 \
--max_face 50 --max_edge 30
python ldm.py --data data_process/abc_parsed \
--list data_process/abc_data_split_6bit.pkl --option edgez \
--surfvae proj_log/abc_vae_surf.pt --edgevae proj_log/abc_vae_edge.pt --gpu 0 1 \
--env abc_ldm_edgez --train_nepoch 300 --batch_size 64 \
--max_face 50 --max_edge 30
### Training Furniture Latent Diffusion Model (classifier-free) ###
python ldm.py --data data_process/furniture_parsed \
--list data_process/furniture_data_split_6bit.pkl --option surfpos --gpu 0 1 \
--env furniture_ldm_surfpos --train_nepoch 3000 --test_nepoch 200 --save_nepoch 200 \
--max_face 50 --max_edge 30 --cf
python ldm.py --data data_process/furniture_parsed \
--list data_process/furniture_data_split_6bit.pkl --option surfz \
--surfvae proj_log/furniture_vae_surf.pt --gpu 0 1 \
--env furniture_ldm_surfz --train_nepoch 3000 --batch_size 256 \
--max_face 50 --max_edge 30 --cf
python ldm.py --data data_process/furniture_parsed \
--list data_process/furniture_data_split_6bit.pkl --option edgepos \
--surfvae proj_log/furniture_vae_surf.pt --gpu 0 1 \
--env furniture_ldm_edgepos --train_nepoch 1000 --batch_size 64 \
--max_face 50 --max_edge 30 --cf
python ldm.py --data data_process/furniture_parsed \
--list data_process/furniture_data_split_6bit.pkl --option edgez \
--surfvae proj_log/furniture_vae_surf.pt --edgevae proj_log/furniture_vae_edge.pt --gpu 0 1 \
--env furniture_ldm_edgez --train_nepoch 1000 --batch_size 64 \
--max_face 50 --max_edge 30 --cf