mirror of
https://github.com/PrimitiveAnything/PrimitiveAnything.git
synced 2025-09-18 05:22:48 +08:00
53 lines
1.1 KiB
YAML
Executable File
53 lines
1.1 KiB
YAML
Executable File
dataset:
|
|
name: base
|
|
pc_dir: ./data/test_pc
|
|
bs_dir: data/basic_shapes_norm
|
|
max_length: 144
|
|
range_scale: [0, 1]
|
|
range_rotation: [-180, 180]
|
|
range_translation: [-1, 1]
|
|
rotation_type: euler
|
|
pc_format: pn
|
|
model:
|
|
attn_depth: 6
|
|
attn_heads: 6
|
|
bin_smooth_blur_sigma: -1
|
|
bs_pc_dir: data/basic_shapes_norm_pc10000
|
|
coarse_pre_gateloop_depth: 3
|
|
continuous_range_rotation:
|
|
- -181
|
|
- 181
|
|
continuous_range_scale:
|
|
- 0
|
|
- 1
|
|
continuous_range_translation:
|
|
- -1
|
|
- 1
|
|
dim: 768
|
|
dim_rotation_embed: 16
|
|
dim_scale_embed: 16
|
|
dim_translation_embed: 16
|
|
dim_type_embed: 48
|
|
dropout: 0.0
|
|
embed_order: ctrs
|
|
gateloop_use_heinsen: false
|
|
loss_weight:
|
|
eos: 1.0
|
|
reconstruction: 1.0
|
|
rotation: 1.0
|
|
scale: 1.0
|
|
translation: 1.0
|
|
type: 1.0
|
|
max_primitive_len: 144
|
|
name: discrete
|
|
num_discrete_rotation: 181
|
|
num_discrete_scale: 128
|
|
num_discrete_translation: 128
|
|
num_type: 3
|
|
shape_cond_with_cat: true
|
|
shape_cond_with_cross_attn: false
|
|
shape_cond_with_film: false
|
|
shape_condition_dim: 768
|
|
shape_condition_len: 77
|
|
shape_condition_model_type: michelangelo
|