torchgeo
torchgeo copied to clipboard
All trainers should compute confusion matrices on at least the test split
- [ ] Add torchmetrics confusion matrix to all trainers
- [ ] Save the test results to a file in the experiment directory (end of
train.py)