{
  "library_name": "torch-pointcloud",
  "name": "act.dvae.runpei-dong",
  "task": "pretraining",
  "hparams": {
    "in_channels": 0,
    "image_dim": 768,
    "image_depth": 12,
    "image_num_heads": 12,
    "num_prompt_tokens": 64,
    "num_groups": 64,
    "group_size": 32,
    "encoder_dims": 384,
    "token_local_channels": [
      128,
      256
    ],
    "token_global_channels": [
      512
    ],
    "num_tokens": 8192,
    "tokens_dims": 384,
    "decoder_dims": 384,
    "prompt_dropout": 0.1
  },
  "input_keys": [
    "x",
    "pos",
    "batch"
  ],
  "dataset": "shapenet55",
  "author": "runpei-dong",
  "license": "MIT"
}