georgeNakayama commited on
Commit
cb5ca61
·
verified ·
1 Parent(s): fc53842

Remove pgf_text/config.yaml (configs live in the code repo)

Browse files
Files changed (1) hide show
  1. pgf_text/config.yaml +0 -138
pgf_text/config.yaml DELETED
@@ -1,138 +0,0 @@
1
- dataset:
2
- normalization:
3
- pcd_mean:
4
- - 1.02373453
5
- - 0.53175375
6
- - 0.0537376513
7
- - 86.1824935
8
- - -1.33205849
9
- - 0.0
10
- pcd_std:
11
- - 0.51201646
12
- - 0.13567009
13
- - 17.43921292
14
- - 36.19954752
15
- - 9.72246543
16
- - 1.0
17
- xlims:
18
- - -150
19
- - 150
20
- ylims:
21
- - -80
22
- - 220
23
- _target_: datasets.garment_particle_v2_dataset_w_text.GarmentParticleDatasetV2WithText
24
- data_dir: /scratch/m000051/george/gcdv2/garment_particles_v2.1_11182025
25
- multiplier: 1.0
26
- n_samples: null
27
- n_points: 8192
28
- fixed_n_points: false
29
- pad_everything: false
30
- text_path: /scratch/m000051/george/gcdv2/short_captions_v2.json
31
- tokenizer_name: openai/clip-vit-large-patch14
32
- max_length: 77
33
- text_drop_prob: 0.1
34
- use_all_captions: false
35
- collate_fn:
36
- _target_: datasets.dynamic_sampler.ConstantTokenCollator
37
- model:
38
- _target_: models.sparse_lightningdit_v3_cross_attn_varlen.SparseLightningDiTV3CrossAttnVarlen
39
- depth: 28
40
- hidden_size: 1152
41
- num_heads: 16
42
- use_qknorm: true
43
- use_swiglu: true
44
- use_rope: false
45
- use_face_emebds: false
46
- use_rmsnorm: true
47
- wo_shift: false
48
- in_channels: 6
49
- use_checkpoint: false
50
- backend: flash-attn
51
- num_face_classes: 34
52
- text_encoder_name: openai/clip-vit-large-patch14
53
- block_type: v1
54
- transport:
55
- _target_: transport.transport.Transport
56
- path_type: Linear
57
- model_type: velocity
58
- loss_type: null
59
- sample_eps: null
60
- train_eps: null
61
- use_cosine_loss: false
62
- use_lognorm: true
63
- denormalize_loss: false
64
- use_closure_loss: false
65
- use_valid_loss: false
66
- sample:
67
- mode: ODE
68
- sampling_method: euler
69
- atol: 1.0e-06
70
- rtol: 0.001
71
- reverse: false
72
- likelihood: false
73
- num_sampling_steps: 250
74
- cfg_scale: 0
75
- per_proc_batch_size: 5
76
- fid_num: 5
77
- cfg_interval_start: 0.11
78
- timestep_shift: 0.3
79
- curve_sampling: false
80
- stitch_sampling: false
81
- ckpt_path: null
82
- data:
83
- val_split: assets/garment_particle_v2_test_11182025.txt
84
- train_split: assets/garment_particle_v2_train_11182025.txt
85
- workers_per_gpu: 8
86
- sampler:
87
- _target_: datasets.dynamic_sampler.ConstantTokenBatchSampler
88
- target_tokens_per_batch: 16382
89
- shuffle: true
90
- token_counts_cache: assets/token_counts_v2.2.pkl
91
- max_batch_size: 128
92
- min_batch_size: 1
93
- bucket_size_bins: 10
94
- drop_last: true
95
- verbose: true
96
- train:
97
- max_steps: 320000
98
- global_batch_size: 64
99
- global_seed: 0
100
- deterministic: false
101
- diff_seed: true
102
- exp_name: xl_w_text_large
103
- wandb_dir: /scratch/m000051/george/InteractGarment
104
- wandb_cache_dir: /scratch/m000051/george/InteractGarment/cache
105
- wandb_project: interact-garment
106
- ckpt: null
107
- log_every: 10
108
- ckpt_every: 1000
109
- sample_every: 5000
110
- eval_every: 10000000000
111
- gradient_accumulation_steps: 1
112
- resume: /scratch/m000051/george/InteractGarment/src/outputs/xl_w_text_large_qknorm/checkpoints
113
- max_checkpoints: 5
114
- mp_start_method: fork
115
- cudnn_benchmark: false
116
- log_level: INFO
117
- eval:
118
- sample_per_batch: 1
119
- n_samples: 4
120
- n_point_list: assets/n_points.txt
121
- caption_list: assets/example_captions.txt
122
- use_ema: true
123
- using_cfg: false
124
- mode: val
125
- optimizer:
126
- lr: 0.0001
127
- beta2: 0.99
128
- max_grad_norm: 1.0
129
- warmup_ratio: 0.01
130
- warmup_steps: 100
131
- weight_decay: 0
132
- dist_params:
133
- backend: nccl
134
- timeout: 3600
135
- fsdp:
136
- param_dtype: bfloat16
137
- reduce_dtype: float32
138
- reshard_after_forward: true