Hugging Face's logo

ADSKAILab
/
WaLa-SK-1B

Image-to-3D
English
wala
sketch-to-3d
Model card Files Community
WaLa-SK-1B
File size: 4,969 Bytes 
65f651c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
 
{
    "max_failures": -1,
    "strategy": null,
    "gpu_workers": 8,
    "restore_path": null,
    "prefetch_factor": null,
    "matual_precision": "medium",
    "max_concurrency": 1500,
    "multipart_size": 8388608,
    "print_every": 100,
    "save_every": 100,
    "validation_every": null,
    "validation_every_log": 500,
    "visualization_every": 15000,
    "log_level": "info",
    "experiment_type": "max",
    "experiment_every": 5,
    "fast_dev_run": false,
    "limit_val_batches": 0.0,
    "ft_train_number": null,
    "ft_train_datasets": null,
    "val_cnt": null,
    "use_even_val": false,
    "use_ema": true,
    "ema_decay": 0.9999,
    "use_ema_weights": true,
    "use_compile": true,
    "batch_size": 8,
    "test_batch_size": 8,
    "num_workers": 23,
    "test_threshold": 0.05,
    "num_gpus": 8,
    "seed": 1,
    "epochs": 300,
    "optimizer": "Adam",
    "lr": 0.0001,
    "train_mode": "train",
    "use_local_storage": false,
    "auto_exp_name": null,
    "input_type": "Wavelet",
    "output_type": "Wavelet",
    "encoder_type": "General_Encoder_Down_2",
    "decoder_type": "General_Decoder_Up_2",
    "encoder_num_tran": 0,
    "decoder_num_tran": 0,
    "last_feature_transform": null,
    "reconstruct_loss_type": "mean",
    "quantizer_type": "original",
    "normalize_latent": null,
    "e_dim": 4,
    "n_e": 1024,
    "beta": 0.25,
    "sample_mode": "bilinear",
    "padding": 0.1,
    "gamma": 1,
    "grid_size": 12,
    "t_loss": 1.0,
    "num_latent_tokens": 256,
    "dataset_name": "all",
    "voxel_transform": null,
    "num_points": 2048,
    "num_sdf_points": 5000,
    "categories": null,
    "resolution": 256,
    "max_depth": 3,
    "max_training_level": 2,
    "point_num": 16384,
    "keep_level": 2,
    "data_keep_level": 2,
    "wavelet": "bior6.8",
    "padding_mode": "constant",
    "use_normalization": false,
    "use_shift_mean": false,
    "start_stage": 0,
    "use_adaptive_stage_update": false,
    "no_rebalance_loss": true,
    "use_compact_indices": true,
    "sample_threshold_ratio": 0.03125,
    "use_batched_threshold": true,
    "use_sample_training": true,
    "use_sample_threshold": true,
    "div_hyp": 1.0,
    "checkpoint": null,
    "use_timestamp": false,
    "num_iterations": 300000,
    "gpu": "0",
    "threshold": 0.45,
    "sampling_type": null,
    "auto_precision": "bf16",
    "gradient_clip_val": 1.0,
    "dropout": 0.0,
    "sdf_points": 20000,
    "sdf_sample_type": "mixture",
    "sdf_res": 256,
    "greater_or_no": true,
    "use_s3": true,
    "wavelet_transform": "all",
    "test_file_name": "model_performance.csv",
    "network_type": "latent_uvit",
    "diffusion_beta_schedule": "cosine",
    "diffusion_step": 1000,
    "diffusion_rescale_timestep": 100,
    "diffusion_scale_ratio": 1.0,
    "diffusion_model_var_type": "FIXED_SMALL",
    "diffusion_model_mean_type": "START_X",
    "diffusion_loss_type": "MSE",
    "diffusion_sampler": "second-order",
    "dit_block_type": "cross_dit",
    "att_patch_size": 1,
    "att_hidden_size": 1152,
    "transformer_num_blocks": 32,
    "transformer_num_heads": 16,
    "transformer_add_num_register": 0,
    "unet_model_channels": 128,
    "unet_num_res_blocks": [
        3
    ],
    "learnable_skip_r": null,
    "add_condition_res_ch": 128,
    "with_fix_pos": true,
    "cond_num_mapping_layers": 0,
    "add_condition_time_ch": true,
    "add_condition_input_ch": null,
    "use_pointcloud_conditions": false,
    "pc_encoder_type": "PointNet_Simple",
    "use_pointvoxel_encoder": false,
    "num_pc_points": 2500,
    "use_pc_samples": false,
    "sample_num": 2500,
    "pc_dims": 1024,
    "num_inds": 1024,
    "pc_output_dim": 1024,
    "use_voxel_conditions": false,
    "voxel_context_dim": 1024,
    "voxel_dim": 8,
    "voxel_resolution": 16,
    "use_image_conditions": true,
    "use_camera_index": false,
    "render_resolution": 384,
    "max_images_num": 55,
    "image_transform": "sketch_dino",
    "clip_model_type": "dino-l-14_reg",
    "input_view_cnt": 4,
    "use_multiple_views_grids": false,
    "training_views": null,
    "testing_views": null,
    "use_depth_conditions": false,
    "use_wavelet_conditions": false,
    "dp_cond": 0.05,
    "scale": 3,
    "guidance_type": null,
    "dp_cond_type": null,
    "use_autoencoder": false,
    "checkpoint_type": "last_ft_10000",
    "pre_quant": true,
    "latent_normalization": false,
    "use_autoencoder_ema": false,
    "filter_path": "filter_list_balanced",
    "finetune": false,
    "finetune_dp_cond": null,
    "finetune_dp_cond_type": null,
    "test_log_num": null,
    "precision": "bf16",
    "gradient_clip_val_2": 1.0,
    "wavelet_transform_2": null,
    "weight_decay": 0.0,
    "opt_eps": 1e-08,
    "latent_lr": 0.0001,
    "pin_memory": false,
    "n_px": 224,
    "cond_emb_dim": 1024,
    "cond_grid_size": 256,
    "cond_grid_emb_size": 1024,
    "condition_dim": 1024,
    "num_cond_vectors": 256
}