Load Custom Pre-Trained PyTorch MLP as Surrogate Model

[vinkyc]

Hi everyone,

I have developed a custom Multi-Layer Perceptron (MLP) model using PyTorch, which has been pre-trained for my specific application. The model architecture has an input size of 579 features and produces an output of size 7. Additionally, I've incorporated dropout layers to enable probabilistic outputs. I would like assistance with loading this model and using it as a surrogate model.

Any guidance on how to implement this would be appreciated!

Thank you!