FiPhi-NeuralMark-V3 / pytorch_model.bin
TejAndrewsACC's picture
Create pytorch_model.bin
388fb24 verified
# Saving the model's weights
torch.save(model.state_dict(), 'pytorch_model.bin')