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

[keerti-ds]

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'

UPSNet/upsnet/models/resnet_upsnet.py

Line 90 in aa8434e

res2, res3, res4, res5 = self.resnet_backbone(data['data'])

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