| Epoch | Accuracy (%) | Loss |
|---|---|---|
| 1 | 39.44 | 1.0778 |
| 2 | 39.48 | 1.0774 |
| 3 | 39.04 | 1.0759 |
| 4 | 38.97 | 1.0781 |
| 5 | 39.44 | 1.0768 |
| 6 | 39.99 | 1.0791 |
| 7 | 39.62 | 1.0771 |
| 8 | 39.37 | 1.0770 |
| 9 | 40.42 | 1.0803 |
| 10 | 38.79 | 1.0780 |
| 11 | 38.75 | 1.0785 |
| 12 | 39.30 | 1.0753 |
| 13 | 39.55 | 1.0792 |
| 14 | 40.13 | 1.0761 |
| 15 | 39.04 | 1.0764 |
| 16 | 40.06 | 1.0759 |
| 17 | 39.01 | 1.0776 |
| 18 | 39.30 | 1.0763 |
| 19 | 40.06 | 1.0772 |
| 20 | 40.06 | 1.0763 |
1from torchvision import models
2import torch
3
4model = models.alexnet()
5model.fc = torch.nn.Linear(model.fc.in_features, 3)
6model.load_state_dict(torch.load('pytorch_model.bin'))
7model.eval()