Relative Content

Tag Archive for pytorchneural-networkactivation-function

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: