How to label a picture and please help me to fix error in my post

I’am senior and my topic is handwritten digit recognition, but i use my dataset to train, not use MNIST dataset. I’m trying follow to this website https://nextjournal.com/gkoehler/pytorch-mnist to train my dataset. Unfortunately, I dont know how to label my image and I have a problem with my code like this. My code and my error below. Thank you everyone!!

https://drive.google.com/drive/folders/1CbkZ7mz2KDrK7MClb5sAOlglY2Z3vlAk?usp=sharing

The easiest way would probably be to store each image in the corresponding subfolder and use an ImageFolder to automatically generate the labels.

Let me try, thank u :smiley: