torch.load
torch.load(f, map_location=None, pickle_module=pickle, , weights_only=False, *pickle_load_args)
Loads an object saved with "torch.save()" from a file.
"torch.load()" uses Python's unpickling facilities but treats
   storages, which underlie tensors, specially. They are first
   deserialized on the CPU and are then moved to the device they were
   saved from. If this fails (e.g. because the run time system doesn't
   have certain devices), an exception is raised. However, storages
   can be dynamically remapped to an alternative set of devices using
   the "map_location" argument.
If "map_location" is a callable, it will be called once for each
   serialized storage with two arguments: storage and location. The
   storage argument will be the initial deserialization of the
   storage, residing on the CPU. Each serialized storage has a
   location tag associated with it which identifies the device it was
   saved from, and this tag is the second argument passed to