File size: 141 Bytes
6d8bcb8 |
1 2 3 4 5 6 |
import torch
# Assuming `model` is your trained model
model_save_path = "pytorch_model.bin"
torch.save(model.state_dict(), model_save_path)
|
6d8bcb8 |
1 2 3 4 5 6 |
import torch
# Assuming `model` is your trained model
model_save_path = "pytorch_model.bin"
torch.save(model.state_dict(), model_save_path)
|