Pytorch
一位以泪洗面的同学
这个作者很懒,什么都没留下…
展开
-
PyTorch 修炼篇(二) : CIFAR-10 分类模型
通过VGG进行cifar-10分类:import osimport timeimport torchimport torch.nn as nnimport torch.optim as optimimport torch.nn.functional as Ffrom torch.utils.data.dataloader import DataLoaderimport torchv...原创 2020-03-28 19:05:16 · 698 阅读 · 0 评论 -
PyTorch 修炼篇(一) : CIFAR-10 分类模型
import osimport timeimport torchimport torch.nn as nn # 神经网络工具箱import torch.nn.functional as F # functional中的函数是一个确定的不变的运算公式,输入数据产生输出就ok。import numpy as npfrom torch.autograd import Variablef...原创 2020-03-04 14:05:57 · 2040 阅读 · 3 评论