Results
2
comments of
Waqwoya Mesfin
The latest CIFAR-10 dataset calls it's labels "targets". So replace line 173 to. `labels = np.array(dataset.targets)`
I'm not sure what's causing it but the problem seems to occur whenever using efficientnet, mobilenet or custom implemented resnets. It works fine when using the torch.models.resnet models.