stylegan2-pytorch logging the losses

logging the losses

Open harsh-agarwal opened this issue 4 years ago • 5 comments

Thanks for the amazing work! Is there any way I could look at the logged losses?

Regards Harsh

Nov 11 '20 20:11 harsh-agarwal

@harsh-agarwal hi! sure! I was thinking perhaps an integration with https://github.com/aimhubio/aim ?

Nov 11 '20 20:11 lucidrains

@harsh-agarwal would just the G and D losses be good enough for you to begin with?

Nov 11 '20 20:11 lucidrains

yes! May be just G, D and GP.

PS: I don't see PL in my logs am I doing something wrong?

Nov 12 '20 04:11 harsh-agarwal

PL kicks in around 5000 steps (at least, it should)

I'll see what I can do!

Nov 12 '20 05:11 lucidrains

https://github.com/lucidrains/stylegan2-pytorch/tree/1.5.0#logging-to-experiment-tracker let me know if that works for you! @harsh-agarwal

Nov 13 '20 22:11 lucidrains