Show HN: Minimal image regression template for PyTorch

Last year I got interested in image regression but struggled to find any easy tutorials or templates for getting set up with image regression in pytorch. Seems like all the tutorials were about image classification.. I decided to make a very minimal pytorch image regression example for a task that I thought would be easy to solve to make it easier for people to clone and then modify with their own datasets. Its pretty basic and has minimal dependencies. Let me know if you spot any bugs or simple improvements :)