Also include torchmetric's ConfusionMatrix. Create a `test_step` PyTorch-Lightning-style to handle this at the end. Logging the output of this can be a bit tricky. - Githubissues

faris-k / fastsiam-wafers

Self-Supervised Representation Learning of Wafer Maps with FastSiam

MIT License

6 stars 0 forks source link

Also include torchmetric's ConfusionMatrix. Create a `test_step` PyTorch-Lightning-style to handle this at the end. Logging the output of this can be a bit tricky. #7

Closed faris-k closed 1 year ago