2022年8月27日
深度学习CNN网络，MNIST数据分类
摘要： import torch import torch.nn as nn import torchvision.datasets as dsets import torchvision.transforms as transforms from torch.autograd import Variabl 阅读全文
posted @ 2022-08-27 21:55 kuanleung 阅读(36) 评论(0) 推荐(0)
BatchNorm和Dropout的使用
摘要： import torch m = torch.nn.BatchNorm1d(10) m = torch.nn.BatchNorm1d(10,affine=False) input = torch.autograd.Variable(torch.randn(20,10)) output = m(inp 阅读全文
posted @ 2022-08-27 14:58 kuanleung 阅读(49) 评论(0) 推荐(0)