pytorch学习
小小程序天
Nothing is impossible.
展开
-
pytorch实现全连接网络的vae
网络全用全连接层nn.Linear()import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transformsimport torch.optim as optimimport osclass VAE(nn.Module): def __init__(self): super(VAE, self).__init__() self.E_f原创 2020-10-25 16:46:59 · 998 阅读 · 0 评论 -
pytorch实现训练Cifar10
import torch.nn as nnimport torch.nn.functional as Fimport torchvisionimport torchvision.transforms as transformsimport torch.optim as optimimport torchclass LeNet(nn.Module): # 构造方法里面定义可学...原创 2020-04-09 09:41:57 · 1117 阅读 · 0 评论 -
pytorch实现LeNet网络结构
import torch.nn as nnimport torch.nn.functional as Fimport torchimport torch.optim as optimclass LeNet(nn.Module): # 构造方法里面定义可学习参数的层 def __init__(self): super(LeNet, self).__init...原创 2020-04-09 09:39:47 · 225 阅读 · 0 评论