Hi, @ptrblck. I actually figured out a way to share the model without disclosing my custom architecture. I’m basing my model on the ITracker model (https://github.com/CSAILVision/GazeCapture/blob/master/pytorch/ITrackerModel.py) which is publicly available on the repo. https://github.com/CSAILVision/GazeCapture/tree/master/pytorch 1. This model suffers from the same issues as mine.
I was able to pin-point that the change which introduced the performance issues was introduced in the torch version 1.3.0.