search

zhanghang1989 / ResNeSt

ResNeSt: Split-Attention Networks
https://arxiv.org/abs/2004.08955
Apache License 2.0
3.23k stars 496 forks source link

RuntimeError: Module 'SplAtConv2d' has no attribute 'dropblock' : #176

Closed vivekkhandelwal1 closed 2 years ago

vivekkhandelwal1 commented 2 years ago

Hi, while running the model, I am getting a runtime error as follows:

RuntimeError: 
Module 'SplAtConv2d' has no attribute 'dropblock' :
  File "/home/vivek/.cache/torch/hub/zhanghang1989_ResNeSt_master/resnest/torch/models/splat.py", line 56
            x = self.bn0(x)
        if self.dropblock_prob > 0.0:
            x = self.dropblock(x)
                ~~~~~~~~~~~~~~ <--- HERE
        x = self.relu(x)

@zhanghang1989, could you please tell me how to resolve this?

Python Version: 3.8.10 Pytorch Version: 1.12.0.dev20220314+cpu

zhanghang1989 commented 2 years ago

Could you comment out that line? dropblock is not needed for inference