Inference error #10
Description
Hi, i trained my own datasets using Efficient_B4, and
instruction as blow:
python inference.py data/custom/test/ --model efficientnet_b4 --img-size 300 -b 32 -j 4 --num-classes 2 --checkpoint output/model_best_b4.pth.tar
The error was
Traceback (most recent call last):
File "inference.py", line 122, in <module>
main()
File "inference.py", line 114, in main
filenames = loader.dataset.filenames()
AttributeError: 'PrefetchLoader' object has no attribute 'dataset'
How to fix it? Thanks a lot!