deepsafe's picture
Add stripped inference-only model code mirror
9e14838 verified
Raw History Blame Contribute Delete
185 Bytes
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)