torch.load — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.load(f, map_location=None, pickle_module=pickle, **pickle_load_args) [source] 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.