Last chance! 50% off unlimited learning
Sale ends in
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 nn.Parameter
classes from the tensors in the dict.
load_state_dict(path)to the state dict file
a named list of tensors.
The above might change with development of this in pytorch's C++ api.