fxmeng/RMNet

AttributeError: 'RMNeXt' object has no attribute 'layer0'

Closed this issue · 2 comments

wolfworld6 commented

when I use the code to train the AE rmnext41_64x5_g16,i got this error, why? how do I solve it?

fxmeng commented

Thanks for your question,
I have changed the stage0 in lines 146&148 to layer0:

RMNet/models/rmnext.py

Line 146 in 2b6175d

self.stage0 = nn.Sequential(*[RepBlock(3, self.in_planes, stride=2), RepBlock(self.in_planes, self.in_planes, stride=2)])

wolfworld6 commented

stage0

thx very much , I know!