Solved by ptrblck in post #2 
              
              
                There are several options to remove/replace this layer depending on your use case. 
You could copy the source code of the model, remove the layer, change the forward method, and store this new model definition as a custom model. This would allow you to load this new model in your scripts and just tr…