As title mentioned, I get OOM error when my model reach the first meta epoch test set phase in calculating anomaly score.
I encounter this error whether using resnet18, DeiT, CaiT.
Seem likes @AlessioGalluccio had try to modify calculate anomaly score into the for loop, but still have the problem.
I'm using GTX1080 8GB.
Is there have any solutions? except decrease the batch size
Duplicate of #9
As title mentioned, I get OOM error when my model reach the first meta epoch test set phase in calculating anomaly score. I encounter this error whether using resnet18, DeiT, CaiT. Seem likes @AlessioGalluccio had try to modify calculate anomaly score into the for loop, but still have the problem. I'm using GTX1080 8GB. Is there have any solutions? except decrease the batch size