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?
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.
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?