SQLGenerator / pytorch_model.bin
Kasivs's picture
Create pytorch_model.bin
6d8bcb8 verified
Raw
History Blame Contribute Delete
141 Bytes
import torch
# Assuming `model` is your trained model
model_save_path = "pytorch_model.bin"
torch.save(model.state_dict(), model_save_path)