Save/Load/resume training?

[jarlva]

Hello and thanks for sharing your code!
Can you please let me know if there is a way to save model state/weights in order to test later or resume training?

[germain-hug]

Hi and thanks for your interest!
I just modified main.py to save weights when done training. I also added a separate script (load_and_run.py) for visualization after training.

[jarlva]

Great! I'll give it a shot