Solved by ptrblck in post #4 
              
              
                How did you save the model? 
This workflow works for me: 
In PyTorch 0.3.1.post3: 
model = nn.BatchNorm1d(3)
torch.save(model.state_dict(), 'test.pt')

In PyTorch 0.4.0: 
model = nn.BatchNorm1d(3)
model.load_state_dict(torch.load('test.pt'))
x = torch.randn(1, 3, 2)
model(x)

Did you save the comple…