arthurdjn's picture
Update README.md and config.json and model.safetensors for act.scanobjectnn-objonly.runpei-dong
f7a3234 verified
Raw History Blame Contribute Delete
927 Bytes
{
"library_name": "torch-pointcloud",
"name": "act.scanobjectnn-objonly.runpei-dong",
"task": "classification",
"hparams": {
"in_channels": 0,
"num_classes": 15,
"embed_dim": 384,
"depth": 12,
"num_heads": 6,
"num_groups": 128,
"group_size": 32,
"encoder_local_channels": [
128,
256
],
"encoder_global_channels": [
512
],
"pos_embed_channels": [
128
],
"drop_path": 0.1,
"dropout": 0.5,
"act": "gelu",
"spatial_dim": 3
},
"input_keys": [
"x",
"pos",
"batch"
],
"dataset": "scanobjectnn-objonly",
"metrics": {
"OA": 91.22,
"mAcc": 90.26
},
"classes": [
"bag",
"bin",
"box",
"cabinet",
"chair",
"desk",
"display",
"door",
"shelf",
"table",
"bed",
"pillow",
"sink",
"sofa",
"toilet"
],
"author": "runpei-dong",
"license": "MIT"
}