![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
csliudh
这个作者很懒,什么都没留下…
展开
-
pytorch显存越来越多的一个潜在原因-- 这个函数还没有在torch.cuda.Tensor中定义
最近在用pytorch跑实验,有如下操作需要用到: (pytorch版本为0.3.1) class SpatialFilter(nn.Module): def __init__(self,mode=True,sf_rate=0.8): ''' 给定特征图求hot_map ''' super(SpatialFilter,se...原创 2018-08-30 11:58:27 · 9836 阅读 · 1 评论 -
使用神经网络拟合argmax函数
最近比较闲,听说神经网络可以拟合任意函数,因此尝试使用神经网络拟合argmax函数。 我使用的是pytorch,就是堆叠多层感知机来拟合一个argmax函数。为了防止不收敛,我的输入是一个 20维的向量,要求输出的值是 0~1 范围之内的,0表示第0个数是最大的,1表示第19个数是最大的,0.4表示第15个数是最大的,可以用 int(输出*20) 转回下标。代码如下: import tor...原创 2018-11-14 22:43:27 · 5155 阅读 · 3 评论