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