@nima_rafiee I was trying to look into why the existing code didn’t work with BatchNorm.  In the code you provided, there is the following import statement:
from resent1D import ResNet1D

Where can I find the resnet1D module?