File size: 185 Bytes
9e14838 | 1 2 3 4 5 6 7 8 9 10 | import torch
def torch_save(arr,file):
if torch.__version__>="1.6.0":
torch.save(arr, file, _use_new_zipfile_serialization=False)
else:
torch.save(arr, file)
|
9e14838 | 1 2 3 4 5 6 7 8 9 10 | import torch
def torch_save(arr,file):
if torch.__version__>="1.6.0":
torch.save(arr, file, _use_new_zipfile_serialization=False)
else:
torch.save(arr, file)
|