Can not run training example

[Black-Smith-Of-Fire]

There seems to be a lot of issues regarding the dependencies in this project. I hope the maintainer of this project fixes this.

Also please provide steps where we can fine tune this model using our own custom dataset.