Training on Colab, how to pass arguments to the .ipynb on colab

[MounirB]

Hello,

Following your advices, I am going to train a model on Colab.
However, I am currently facing a small problem, I don't know how to pass arguments to my .ipynb.
Can you tell me how you did it ?

By the way, do I need to convert all the .py files called by the train script to .ipynb or is it enough to have only the main file with .ipynb extension (train.py, in this case) ?

Thank you for your help !

[MounirB]

I added myself the arguments this way after the FLAGS definitions :

FLAGS.pipeline_config_path = './pipeline.config'
FLAGS.train_dir = './train'

Am I right ?

As It is training, I think that my problem is somehow solved. I'll close the issue when I'll get a trained model