File size: 81 Bytes
388fb24 |
1 2 3 |
# Saving the model's weights
torch.save(model.state_dict(), 'pytorch_model.bin')
|
388fb24 |
1 2 3 |
# Saving the model's weights
torch.save(model.state_dict(), 'pytorch_model.bin')
|