深度学习打卡1
"""
使用Mnist数据集来测试上节使用的cross entropy
"""
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets, transforms
#1、设置网络超参数
batch_size=200
learning_rate=0.01
epochs=10
#2、从网上下载MNI.
原创
2021-04-05 15:30:41 ·
207 阅读 ·
0 评论