Bug for SegNet Maybe
Closed this issue · 1 comments
holmosaint commented
Hi:) It seems that there are 2 bugs in your SegNet code. Both of them lie in the _Decoder class in function "init". The out channels of conv2d and batchnorm2d should be 'n_out_feat' rather than 'n_in_feat'.
trypag commented
Nice catch