1 / pytorch_model.bin
rameye's picture
Create pytorch_model.bin
c2c5c38 verified
raw
history blame
214 Bytes
import torch
# Assuming your fine-tuned model is named 'model' (replace it with your model's actual name)
ret_model.pth = ...
# Save the model state dictionary
torch.save(model.state_dict(), 'pytorch_model.bin')