Are there any scripts to train the models ?

[Dorian25]

Dorian, good work in the implementation! Perhaps we can try to train in colab and kaggle environment. Are there any scripts to train the models? Thks.

Originally posted by @pchankh in #2 (comment)

[Dorian25]

Thx mate !
We started the implementation via Colab ! But unfortunately, when we started training the first module, Colab's Ram reached its limit and stopped the process...

For the training, everything is in the Main.py file where you can choose to train the 3 modules (core, color and spatial) separately or one at a time using the booleans train_core, train_color and train_spatial...