- 博客(3)
- 收藏
- 关注
原创 pytorch自定义操作层,正向和反向
torch 自定义操作层, 前向和反向过程 在torch中定义正向一定要注意是否使用Variable 比如,在定义层的forward中 正确定义是这样的: def forward(self, infeature): infeature = infeature.reshape(self.inshape[1], self.inshape[2]) / self.p_pixel_num infeature = torch.mm(infeature, self.pooling)
2020-06-30 09:39:56 754
原创 python 数据中存在nan
python 数据中显示nan nan的表示这个数是个非数,就像 1除0. 所以检查下哪里存在除0的情况,或者有其他的位置存在计算成非数。 nan会随着计算传播至后续计算
2020-06-29 17:21:56 2771
原创 解决AttributeError
解决AttributeError: object has no attribute ‘log_softmax’ 错误例子:net = FCN(data) 直接在构建网络时,给与输入 改为:net = FCN() out = net(data)
2020-06-16 19:42:51 2334 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人