Merge pull request #7 from pushpankar/master

Fix flake8 issues

Add support for finetuning more than 2 classes

Fix python 3 download script and left-over cuda in attention layer

Initial commit