Load a state dict fileSource:
This function should only be used to load models saved in python.
For it to work correctly you need to use
torch.save with the flag:
_use_new_zipfile_serialization=True and also remove all
classes from the tensors in the dict.
The above might change with development of this in pytorch's C++ api.