bentrevett / pytorch-image-classification

Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.
MIT License
978 stars 243 forks source link

Got error on test_data.classes #12

Closed TeTeTang closed 3 years ago

TeTeTang commented 3 years ago

The AttributeError: 'Subset' object has no attribute 'classes' was returned for "output_dim = len(data_process.test_data.classes)"

bentrevett commented 3 years ago

Which cell in which notebook is giving you this error?

TeTeTang commented 3 years ago

The issue was fixed.