摘要: !!!一下训练的样本我是通过mps进行加速了训练过程中,如果使用NVIDIA 显卡需自己进行调整device 通过torchvision.datasets.MNIST 我们可以下载到已经准备好的素材,可以通过一下代码进行训练 import torch import torch.nn as nn im 阅读全文
posted @ 2025-10-08 15:49 方子敬 阅读(17) 评论(0) 推荐(0)