arthurdjn's picture
Update config.json for point-bert-base.dvae.xumin-yu
118fb2f verified
Raw History Blame Contribute Delete
524 Bytes
{
"library_name": "torch-pointcloud",
"name": "point-bert-base.dvae.xumin-yu",
"task": "pretraining",
"hparams": {
"in_channels": 0,
"num_groups": 64,
"group_size": 32,
"encoder_dims": 256,
"token_local_channels": [
128,
256
],
"token_global_channels": [
512
],
"num_tokens": 8192,
"tokens_dims": 256,
"decoder_dims": 256
},
"input_keys": [
"x",
"pos",
"batch"
],
"dataset": "shapenet55",
"author": "xumin-yu",
"license": "MIT"
}