# Saving the model's weights
torch.save(model.state_dict(), 'pytorch_model.bin')