Requirements

[hussius]

Thanks for a nice package! Just wanted to point out that there are more requirements than what is stated in the README file; in addition to the ones already listed:

easydict
matplotlib
sklearn
tensorboardX
torchvision==0.4.2 (not sure exactly which versions work, but this one does and some others I tried don't)

[ayush-angelium]

Try this "0.6.0+cu101" for torchvision.

[Hani1-2]

Did you successfully setup the dataset?
I have an issue can you please help me?