摘要: 载入训练好的模型并且进行预测 代码: import torch import numpy as np import torchvision #torch的视觉包 import torchvision.datasets as datasets import torchvision.transforms 阅读全文
posted @ 2021-03-28 20:35 司砚章 阅读(127) 评论(0) 推荐(0) 编辑
摘要: 利用卷积神经网络对MINIST数据集进行训练 代码: import torch import numpy as np import torchvision #torch的视觉包 import torchvision.datasets as datasets import torchvision.tr 阅读全文
posted @ 2021-03-28 11:19 司砚章 阅读(473) 评论(0) 推荐(1) 编辑
摘要: 多层感知机 代码: import torch import numpy as np import torchvision #torch的视觉包 import torchvision.datasets as datasets import torchvision.transforms as trans 阅读全文
posted @ 2021-03-28 10:40 司砚章 阅读(95) 评论(0) 推荐(0) 编辑
摘要: 对对MNIST数据集进行训练 代码: import torch import numpy as np import torchvision #torch的视觉包 import torchvision.datasets as datasets import torchvision.transforms 阅读全文
posted @ 2021-03-28 10:27 司砚章 阅读(49) 评论(0) 推荐(0) 编辑