arthurdjn's picture
Update README.md and config.json and model.safetensors for act.modelnet40.runpei-dong
0d58e7d verified
Raw History Blame Contribute Delete
1.27 kB
{
"library_name": "torch-pointcloud",
"name": "act.modelnet40.runpei-dong",
"task": "classification",
"hparams": {
"in_channels": 0,
"num_classes": 40,
"embed_dim": 384,
"depth": 12,
"num_heads": 6,
"num_groups": 64,
"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": "modelnet40",
"metrics": {
"OA": 93.19,
"mAcc": 89.9
},
"classes": [
"airplane",
"bathtub",
"bed",
"bench",
"bookshelf",
"bottle",
"bowl",
"car",
"chair",
"cone",
"cup",
"curtain",
"desk",
"door",
"dresser",
"flower_pot",
"glass_box",
"guitar",
"keyboard",
"lamp",
"laptop",
"mantel",
"monitor",
"night_stand",
"person",
"piano",
"plant",
"radio",
"range_hood",
"sink",
"sofa",
"stairs",
"stool",
"table",
"tent",
"toilet",
"tv_stand",
"vase",
"wardrobe",
"xbox"
],
"author": "runpei-dong",
"license": "MIT"
}