search

whai362 / pan_pp.pytorch

Official implementations of PSENet, PAN and PAN++.
Apache License 2.0
439 stars 90 forks source link

Backbone Resnet101 - Training time error #36

Closed suresh-s closed 3 years ago

suresh-s commented 3 years ago

I have changed the backbone to Resnet101 and started the training,

got the following error

RuntimeError: Given groups=1, weight of size 128 64 1 1, expected input[8, 256, 160, 160] to have 64 channels, but got 256 channels instead

Could you tell me the reason for this error?

czczup commented 3 years ago

The out channels of the four stages in R101 are 256, 512, 1024, 2048 respectively. You should set in_channels=(256, 512, 1024, 2048) in the config file.