Hi @Nikronic,
First, thanks a lot for your precious help.
You said I may want to reformat my code using a framework or use a proper IDE/editor.  I am using Jupyter notebook (anaconda software) and the Preview (Nightly) pytorch version.