configs/infer.yml

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