Using PyTorch Vision to build an AI model that classifies and labels AI-generated images with high accuracy
- Traceback (most recent call last): File "", line 1, in File "/Library/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/spawn.py", line 116, in spawn_main exitcode = _main(fd, parent_sentinel) File "/Library/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/spawn.py", line 126, in _main self = reduction.pickle.load(from_parent) AttributeError: Can't get attribute 'CustomImageDataset' on <module 'main' (built-in)>
- Issue with DataLoader Some images in between get shape of (1, 64, 64) instead of (3, 64, 64)