How to freeze BN layers while training the rest of network (mean and var wont freeze)

17

removing this line and only keeping the .eval() solved the problem.