Couldn't we train our model on colab and then download in the exam directory to submit?