Could you add some prints ? Is it working without the Task.init call? the code looks okay and the
- No repository found, message basically says it logs it as a standalone script (which makes sense)
Is this reproducible? I tried to run the same example code on my machine, and it started training ...
Do you have issues with other pytorch examples? Could you try simple reporting example: