Unanswered
Hi Everybody,
I’M Getting Errors With Automatic Model Logging On Pytorch (Running On A Dockered Agent).
Something like:
model = SomePytorchModel()
checkpoint = {'model_state_dict': model.state_dict()}
torch.save(checkpoint, “model.tar”)
161 Views
0
Answers
2 years ago
one year ago