Train all parameters or only BN layers

[fvarno]

As I understand from the paper, you intend to only train the params of the BN layers. Is this what you are trying to do at this line? If so please be aware that module.eval() does not freeze the module’s params. If this is not what you are trying to do please kindly let me know what is the purpose of the mentioned line.

Thanks