config.yaml

model:
  name: charm
  num_discrete_translation_x: [96, 320, 96]
  num_discrete_translation_y: [96, 160, 256]
  num_discrete_translation_z: [96, 320, 96]
  continuous_range_translation_x: [-0.501, -0.1, 0.1, 0.501]
  continuous_range_translation_y: [-0.501, 0.0, 0.3, 0.501]
  continuous_range_translation_z: [-0.501, -0.15, 0.1, 0.501]
  dim_translation_embed: 16
  num_discrete_width: [64, 64]
  continuous_range_width: [0, 0.03, 0.1]
  dim_width_embed: 16
  num_discrete_thickness: [64, 64]
  continuous_range_thickness: [0, 0.02, 0.1]
  dim_thickness_embed: 16
  embed_order: xyzwt
  dim: 768
  attn_depth: 6
  attn_heads: 6
  max_token_len: 8192
  dropout: 0.
  coarse_pre_gateloop_depth: 3
  gateloop_use_heinsen: False
  shape_cond_with_cross_attn: False
  shape_cond_with_film: False
  shape_cond_with_cat: True
  shape_condition_model_type: michelangelo
  shape_condition_len: 77
  shape_condition_dim: 768
dataset:
  name: dev
  pc_dir: ./test_cases/pc
  max_length: 8192
  range_translation: [-0.501, 0.501]
  range_width: [0, 0.1]
  range_thickness: [0, 0.1]
  use_pc: True
  pc_format: pn