Pytorch
PyTorch的实战使用
明曦君
入门的数据价值奋斗者,努力中ing
展开
-
PyTorch实践系列(二):GPU与CPU运行对比
写在开头:本节其实在学习了上一节的MNIST手写体分类后,自己又按照VGG的迷你版搭建了一遍网络,再运行的时候突然想到可以用GPU来跑,于是本节就GPU与CPU运算速度进行一个对比。内容安排今天主要是简单介绍一下VGG16网络架构还有如何将Torch代码放到GPU上运行,然后比较GPU与CPU运行的花费时间,因为采用的是MNIST数据集所以没办法搭建完整的VGG16网络,就按个迷你版的把。...原创 2020-04-01 11:26:40 · 10934 阅读 · 0 评论 -
PyTorch实践系列(一):MNIST 手写体分类
其实我现在还不太懂PyTorch的过多操作,因为是先上手一个项目,那我就尽量将这个项目讲明白,讲不明白的我会在学习一段时间之后回来进行补充。内容安排本次内容我们通过使用PyTorch来进行MNIST手写体分类的一个CNN训练,因为今天花费了一些时间来安装这个PyTorch和TensorFlow,于是兴奋的开始了第一个项目,我们将一边运行代码一边解释的模式进行,在必要部分我们将认真的讲解一下...原创 2020-03-19 14:41:41 · 5749 阅读 · 8 评论