Pytorch学习遇到的问题：AttributeError: module ‘torch.nn’ has no attribute ‘relu’

import torch 
import torch.nn

x = nn.relu(self.fc1(x)) 
# 中间使用了nn.relu函数之后报错
AttributeError: module 'torch.nn' has no attribute 'relu'

之所以会出现这个问题是自己没有弄清楚nn.relu()函数和nn.functional.relu函数的使用方法，nn.relu()作为一个层结构，必须添加到nn.Module容器中才能使用，而nn.functional.relu则作为一个函数调用。
你可以

import torch.nn.functional as F
x = F.relu(self.fc1(x))

这样单独用就不会报错了。解决来源–

# pytorch # 学习 # 深度学习
Logo
向您推荐>>Eolink开发者社区

权威｜前沿｜技术｜干货｜国内首个API全生命周期开发者社区

更多推荐

ELK实现containerd的容器日志采集展示【基于logging的全栈监测】

企业级ELK Stack构建介绍

avatar 云原生

深入理解 Mocha 测试框架：从零实现一个 Mocha

前言什么是自动化测试自动化测试在很多团队中都是Devops环节中很难执行起来的一个环节，主要原因在于测试代码的编写工作很难抽象，99%的场景都需要和业务强绑定，而且写测试代码的编写工作量往往比编写实际业务代码的工作量更多。在一些很多业务场景中投入产出比很低，适合写自动化测试的应该是那些中长期业务以及一些诸如组件一样的基础库。自动化测试是个比较大的概念，其中分类也比较多，比如单元测试，端对端测试，集

avatar 云原生

(20200916 Solved)docker-compose up创建容器自动退出

问题描述如题，创建容器后自动退出了。并且docker start container无效解决方案原因是缺失了控制终端的配置，需要在docker-compose.yml中增加tty:true ，有时候这样也不行，需要再增加一个command:/bin/bash，命令不一定是这个，需要是一个不会退出的命令，然后用-d后台启动容器。Referencesdocker-compose启动容器后自动退出...

avatar 云原生