Solved by ptrblck in post #2 
              
              
                Is conv_batchnorm an nn.BatchNorm layer? 
Was the state_dict saved in another PyTorch version than the one you are using to reload it?