| { | |
| "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" | |
| } | |