Hi all,
I hope that you are having a great day.
I am implementing a paper on uncertainty estimation and using torch-vision pre-trained model ResNet-18.  However I want to use my own customize activation function in the second last layer of resnet-18 instead of relu. How do I do that?
I searched online but found no solution
Thank you.