pytorch入门到实践
文章平均质量分 92
天涯小才
这个作者很懒,什么都没留下…
展开
-
AI打游戏(DL基础)pytorch框架学习笔记
处理数据假如我们要处理的数据有10000张图片,我们既不能一次传进去一万张,这样显存会爆掉,也不能一次只传一张,这样效率太低了。所以我们需要把所有图片分批传进来,比如分100个batch,每个batch里面有100张,也就是batch_size=100。我们训练神经网络的时候并不是把所有数据丢进去训练一轮就行了,而是要反复丢进去学习几轮,这个轮数就叫epoch。计算机视觉方向经常处理的数据是图片,那么我们处理数据后得到的是图片以及它的标签,再将Image+label传入神经网络进行训练。下面我们用伪.原创 2021-05-31 21:58:47 · 1375 阅读 · 2 评论 -
第一课two_layer_neural_net 学习笔记
第一课什么是PyTorch?PyTorch是一个基于Python的科学计算库,它有以下特点:类似于NumPy,但是它可以使用GPU可以用它定义深度学习模型,可以灵活地进行深度学习模型的训练和使用TensorsTensor类似与NumPy的ndarray,唯一的区别是Tensor可以在GPU上加速运算。import torch构造一个未初始化的5x3矩阵:x = torch.empty(5,3)xtensor([[ 0.0000e+00, -8.5899e+09, 0.0000原创 2021-02-24 17:04:45 · 353 阅读 · 0 评论