Training framework & DataSet

[amitsrivastava78]

@becauseofAI , Went though your code and the results of the model are very impressive, we would like to ask you if you have the training framework as well as the dataset as we would like to integrate this and test the model after training and check it accuracy and inference time.

Regards Amit