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