New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
When running DenseNet and MobileNet I got the following errors #2
Comments
|
Face same error too on DenseNet |
I have fixed the issue with these models. Please upgrade the package or pull the latest code to test the changes. I'm closing this issue, feel free to open it up if you face any further issues. |
torchsummary is not working for densenet
|
I also faced this problem until |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
MobileNet:
Traceback (most recent call last):
File "MobileNet.py", line 365, in
summary(model, (3,32,32))
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torchsummary/torchsummary.py", line 56, in summary
model(x)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "MobileNet.py", line 114, in forward
x = self.model(x)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/container.py", line 67, in forward
input = module(input)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/container.py", line 67, in forward
input = module(input)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 359, in call
hook_result = hook(self, input, result)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torchsummary/torchsummary.py", line 28, in hook
params += th.prod(th.LongTensor(list(module.bias.size())))
AttributeError: 'NoneType' object has no attribute 'size'
DenseNet:
Traceback (most recent call last):
File "DenseNet.py", line 237, in
summary(model, (3,32,32))
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torchsummary/torchsummary.py", line 56, in summary
model(x)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "DenseNet.py", line 226, in forward
features = self.features(x)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/container.py", line 67, in forward
input = module(input)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 359, in call
hook_result = hook(self, input, result)
File "/home/sriharsha/miniconda3/lib/python3.6/site-packages/torchsummary/torchsummary.py", line 28, in hook
params += th.prod(th.LongTensor(list(module.bias.size())))
AttributeError: 'bool' object has no attribute 'size'
The text was updated successfully, but these errors were encountered: