深度学习
Hui_R_Q_XIONGDA
这个作者很懒,什么都没留下…
展开
-
CUDA out of memory. Tried to allocate 230.00 MiB (GPU 0; 2.00 GiB total capacity; 1.12 GiB already a
CUDA out of memory. Tried to allocate 230.00 MiB (GPU 0; 2.00 GiB total capacity; 1.12 GiB already allocated; 218.58 MiB free; 1.13 GiB reserved in total by PyTorch) 在使用深度学习验证测试的时候爆显卡了,还不知道解决方法原创 2020-12-02 12:37:11 · 3844 阅读 · 0 评论 -
pytorch官网demo代码详解
model.py # 导入pytorch这个包 import torch.nn as nn import torch.nn.functional as F # 定义一个类,叫LeNeT,来自于nn.Module class LeNet(nn.Module): # 初始化 def __init__(self): # super解决的是在一层一层调用父类时出现的问题,反正每次init的时候加上这句话准没错 super(LeNet, self).__init__(原创 2020-06-09 22:20:45 · 1166 阅读 · 0 评论 -
深度学习(一):卷积神经网络的基本概念
深度学习(一):卷积神经网络的基本概念 1.卷积层 卷积层是卷积神经网络的核心。 什么是卷积?我认为,类似于一个过滤器,或者 如何卷积?通俗来说,用一个滑动窗口在特征中滑动并进行相应的计算——卷积核与之进行相乘,相加,然后strid表示步距,即滑动窗口滑几个格子 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-vJrng99h-1591437324990)(assets/1591436367286.png)] 特性:局部感知;权值共享 1.卷积核的channel与输入特征层的c原创 2020-06-06 17:57:57 · 922 阅读 · 0 评论