getting Pytorch activation function from .pth model
I am looking for a method to retrieve the activation functions used in a PyTorch network saved as a .pth file (torch.save(model)
). Indeed, if the activation functions were not declared in the class when creating the model but only in the forward method, I am unable to identify these activation functions
eg: