Training the model

Thank you for this implementation, great ! I have no (recent) GPU in my computer so I work on Google Colab. In the free version there's a time limit, so I dont think if I can train the model myself. Do you think you will share pre-trained files ? In any case, thank you very much !