Pytorch model function gives Error :'NoneType' object has no attribute 'size'

Question

I trying to run below code in my image segmentation problem without CUDA as I don't have GPU. I have trained my model on CPU using pytorch but on prediction level I'm getting

AttributeError: 'NoneType' object has no attribute 'size'

Here's the code:

idx = 20
model.load_state_dict(torch.load('/content/best_model.pt'))

image, mask = validset[idx]
image = image.unsqueeze_(0)
print(type(image))
# logits_mask = model(image.to(DEVICE).unsqueeze(0)) # (c,h,w) -> (1,c,h,w)
logits_mask = model(image) # (c,h,w) py-> (1,c,h,w)

The resulting error, from the output, is at line number 8:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-56-edf3f0fae49c> in <module>
      6 print(type(image))
      7 # logits_mask = model(image.to(DEVICE).unsqueeze(0)) # (c,h,w) -> (1,c,h,w)
----> 8 logits_mask = model(image) # (c,h,w) py-> (1,c,h,w)
      9 
     10 pred_mask = torch.sigmoid(logits_mask)

3 frames
/usr/local/lib/python3.7/dist-packages/segmentation_models_pytorch/losses/dice.py in forward(self, y_pred, y_true)
     57     def forward(self, y_pred: torch.Tensor, y_true: torch.Tensor) -> torch.Tensor:
     58 
---> 59         assert y_true.size(0) == y_pred.size(0)
     60 
     61         if self.from_logits:

AttributeError: 'NoneType' object has no attribute 'size'

Can you provide the full Traceback? That would help in finding the source of the problem. The error you are seeing is common, and is explained in the answers to this question: stackoverflow.com/questions/8949252/… — Kelly Kuzemchak
– Kelly Kuzemchak, Commented Aug 16, 2022 at 20:33

Riddhiman Dasgupta · Accepted Answer · 2022-08-17 16:24:16Z

1

assert y_true.size(0) == y_pred.size(0) erroring signifies that either y_true or y_pred are None, so you can try checking the types of image, model(image), & mask respectively.

IMO, this might be the root cause: image = image.unsqueeze_(0)

unsqueze_ is an inplace operator, which means it will return nothing and change the tensor image inplace.

answered Aug 17, 2022 at 16:24

Riddhiman Dasgupta

3931 gold badge6 silver badges22 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

bias · Accepted Answer · 2022-08-17 19:34:47Z

0

What is the output of print(type(image))?

Since the error is nonetype for model(image) presumably the issues is with the model load or when you index the image in the following lines:

model.load_state_dict(torch.load('/content/best_model.pt'))

image, mask = validset[idx]

EDIT: What's the output of print(image.size()) and print(image)?

t.size()

edited Aug 17, 2022 at 19:34

answered Aug 16, 2022 at 20:37

bias

1,4663 gold badges20 silver badges40 bronze badges

2 Comments

cs_nerd Over a year ago

Value of print statement is : <class 'torch.Tensor'>

bias Over a year ago

What's the output of print(image.size()) and print(image)?

Stack007 · Accepted Answer · 2022-09-25 07:09:28Z

0

You need to pass the mask as the second parameter to avoid NoneType. # logits_mask = model(image.to(DEVICE).unsqueeze(0), mask(DEVICE).unsqueeze(0)) # (c,h,w) -> (1,c,h,w) #or# logits_mask = model(image, mask) # (c,h,w) py-> (1,c,h,w)

answered Sep 25, 2022 at 7:09

Stack007

1

Collectives™ on Stack Overflow

Pytorch model function gives Error :'NoneType' object has no attribute 'size'

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related