Hi,
I am aware of Apex package. However, I think the problem is not like this.
firstly Feature Extraction Module a network that computes sparse tensor. x0 and x1 are the input for MinkowskiNetwork with coordinates, features. See details here: https://stanfordvl.github.io/MinkowskiEngine/
After Feature Extraction Module, the output has to be extracted into coords and features based based on batch. Then compute detection and description of points then later generate loss.
The problem here is for feature extraction time in forward() only takes 1 second or less than one second for one batch. However, for detection, the forward time is almost half an hour in the same batch, which is definitely unnormal.
As for input concatenation, x0 and x1 is of different size. So it’s not possible to do concatenation at input time. I am looking forward to your reply. Thank you very much!