Legacy autograd function with non-static forward method is deprecated and will be removed in 1.3.

江南酷哥lazy

已于 2022-01-23 18:23:10 修改

阅读量1k

点赞数

分类专栏：一些错误的解决方案文章标签： pytorch 深度学习 python

于 2022-01-23 18:21:18 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_43466026/article/details/122654915

版权

一些错误的解决方案专栏收录该内容

8 篇文章

订阅专栏

在运行pytorch版SSD目标检测网络https://github.com/amdegroot/ssd.pytorch时遇到这个错误
使用Detect类创建对象self.detect执行检测，Detect类继承自Function类，在pytorch1.3及以后的版本需要规定forward方法为静态方法，所以在pytorch1.3以上的版本执行出错。
在这里插入图片描述
解决方法：加forward

        if self.phase == "test":
            # output = self.detect(loc.view(loc.size(0), -1, 4), # 将网络的位置预测表示为（-1，4）形式torch.Size([1, 8732, 4])
            #     self.softmax(conf.view(conf.size(0), -1,self.num_classes)), # 将网络的置信度预测表示为（-1,5）形式torch.Size([1, 8732, 5])
            #     self.priors.type(type(x.data))                  # 先验anchor box，格式（x,y,w,h）
            # )
            ##########################################################           
            output=self.detect.forward(loc.view(loc.size(0), -1, 4),  
                                                self.softmax(conf.view(conf.size(0), 
                                                -1,self.num_classes)),          
                                                                      self.priors.type(type(x.data)) )

thanks