'senet154' pretrained model needs input_channel=3, but your model's input is 4. and I can't find the transformation between the input channel.How you do that ?
Bigwode opened this issue · 3 comments
Bigwode commented
'senet154' model pretrained on imagenet needs input_channel=3, but your model's input is 4. And I can't find the transformation between the input channel.How you do that ?Thanks!
earhian commented
I used try-except for loading pretrained model in line 367 of senet.py.
Bigwode commented
@earhian Oww, It means that the pretrained weights in the first Conv are not loaded in your model, right?
earhian commented
yes!