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)
|