search

microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
https://nni.readthedocs.io
MIT License
14.06k stars 1.82k forks source link

Custom Operations in PointNet++ Implementation #5621

Closed ekurtgl closed 1 year ago

ekurtgl commented 1 year ago

Describe the issue: Hi,

I am trying to implement PointNet++ model with NNI module for NAS, but I am getting the error mentioned in #5610. I was wondering if the custom operations used in pointnet are supported by NNI. You may find the custom ops used in this file:

https://github.com/yanx27/Pointnet_Pointnet2_pytorch/blob/master/models/pointnet2_utils.py

I am not sure if the cause of the error is one of these functions, but if you have any guess on what might be the reason, I can take a look at it. Thank you in advance!

ultmaster commented 1 year ago

If you are not using hardware-aware NAS, the error is unlikely to be correlated to the operations you are using.

ekurtgl commented 1 year ago

Got it, thank you!