Skip to content
Advertisement

How to change activation layer in Pytorch pretrained module?

How to change the activation layer of a Pytorch pretrained network? Here is my code :

JavaScript

Here is my output:

JavaScript

Advertisement

Answer

._modules solves the problem for me.

JavaScript
User contributions licensed under: CC BY-SA
6 People found this is helpful
Advertisement