Using smbclient to load PyTorch model weights from a server, but I encountered some issues. #37
Open
Description
I tried using smbclient to load PyTorch model weights from a server, but I encountered some issues.
file_path = r"\\x.x.x.x\home\pytorch-models\resnet18-f37072fd.pth"
with smbclient.open_file(file_path, mode='rb', buffering=0) as fd:
data = torch.load(fd, map_location=torch.device('cpu'))
The error
Exception has occurred: RuntimeError
PytorchStreamReader failed reading file data/140219036705280: file read failed
Is there any way I can load model weights?
Metadata
Assignees
Labels
No labels