完成模型日志记录模块

Data/loadImage.py

@@ -1,6 +1,12 @@
-from PIL import Image
+'''
+@作者:你遇了我321640253@qq.com
+@文件:loadImage.py
+@创建时间:2023 11 20
 
-from torch.utils.data import Dataset,DataLoader
+模型训练数据集
+'''
+from PIL import Image
+from torch.utils.data import Dataset
 from torchvision.datasets import MNIST
 from torchvision import transforms