Train code

[prinshul]

Thanks for sharing your great work.

Can you please point to the training code in the repo.

Thanks once again.

[Ehteshamciitwah]

Sorry for the delay. test code is provided to validate the model. However, the training file will be shared soon after the paper decision.