Open keerti-ds opened 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
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'
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
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