andreasbinder / Point-GNN-PyTorch

MIT License
6 stars 1 forks source link

How to train the model on KITTI 3D detection dataset and submit the results to KITTI? #2

Closed ZhenhuiTang closed 12 months ago

andreasbinder commented 12 months ago

Hi! Here you can find the input parameters.

I suspect num_node_features will stay the same, you might need to adjust the number of output features num_classes.

Other than that you can instantiate it as any other PyG model. Hope it helps!