I’m using Trainer to train my model.
I have the following outputs on screen:
| Epoch | Training Loss | Validation Loss | Accuracy |
|---|---|---|---|
| 0 | No log | 1.114260 | 0.342667 |
| 1 | No log | 0.939480 | 0.545333 |
| 2 | No log | 0.816581 | 0.660000 |
| 3 | No log | 0.752204 | 0.710667 |
| 4 | No log | 0.741462 | 0.741333 |
| 5 | No log | 0.801005 | 0.754667 |
| 6 | 0.675800 | 0.892765 | 0.748000 |
| 7 | 0.675800 | 1.190328 | 0.752000 |
| 8 | 0.675800 | 1.272624 | 0.745333 |
- Why there are no logs for epochs 0-5 ? (Do I need to configure / enable them ?)
- Epoch #5 got best accuracy.
When I will use predict, which model checkpoint will be used ? (the model which trained after 5 epochs or the model which trained after 8 epochs) ?