sog_unit
sog_unit t1_ixi6ekm wrote
As it is for work choose a framework that will allow you to easily iterate over ideas quickly without having to reinvent the wheel. For hobby projects you can mess around and enjoy the learning. You are better off spending your time on the problems that have a business impact.
I am using lightning for a research project and one thing I like about it is that, if my code runs locally, then I am almost certain that it will run on the cloud. If you need more control there is lightning lite that allows you to write custom loops while it handles much of the device stuff for you.
Whichever method you go with have fun and enjoy the learning process.
sog_unit t1_j0uv5iq wrote
Reply to [D] Resources to learn and fully understand Diffusion Model Codes by Itachi_99
Lucidrains GitHub is always my go to repo for understandable paper implementations https://github.com/lucidrains/denoising-diffusion-pytorch