https://github.com/cs230-stanford/cs230-code-examples/blob/159df10a6187c7e1d6ec949c8e06d7f67f8f1cd2/pytorch/vision/train.py#L51
cs230-code-examples/pytorch/vision/train.py
Line 51 in 159df10