In this HW, you will implement several neural networks using PyTorch.
Please follow the description in this jupyter notebook
In Q4, if you want to use YOLOv6 or later, it is fine as well.