this code snippet works fine and I am able to load the tensor.
So what I do iis that during training I compute batchNorm statistics and I store them using torch.save() and I get no error during this phase but when I came to load it (I forgot to mention this) some tensors are loaded and some are not and it does not depend on the size(sometimes a tensor of size 500 is loaded  and other with size 100 is not)