search

uber-research / UPSNet

UPSNet: A Unified Panoptic Segmentation Network
Other
648 stars 120 forks source link

TypeError: new(): invalid data type 'str' #145

Open keerti-ds opened 2 years ago

keerti-ds commented 2 years ago

Hi,

I would like to get the summary of the model using pytorchsummary. I used below code for the same in upsnet_end2end_train.py

input_shape = [1, 3, 224, 224], summary(train_model, input_shape)

But this throws me an error as below File "../upsnet/models/resnet_upsnet.py", line 90, in forward res2, res3, res4, res5 = self.resnet_backbone(data['data']) TypeError: new(): invalid data type 'str'

https://github.com/uber-research/UPSNet/blob/aa8434e5a721ed217849607815304f68dfd7720a/upsnet/models/resnet_upsnet.py#L90

If someone can guide me what could be the issue? Thanks in advance