2b6f86c 5715641
1
2
3
4
5
6
# train_model.py import torch # Assuming `model` is your trained model torch.save(model.state_dict(), 'mnist_model.pth')
