https://stevenwalton.github.io/posts/VGGP2/
Making VGG with PyTorch: Part 2 - Tyro