I have updated my own project the third time and it should be looking good.
https://dev.to/qbaocaca/my-pytorch-training-pipeline-in-image-classification-dg7