diff --git a/ppcls/arch/backbone/legendary_models/hrnet.py b/ppcls/arch/backbone/legendary_models/hrnet.py index 545925ce6dabf6d13934feb7768c58b08a6ba0c8..c0a7f8509541d8e8a197c998ae207c60a6d646e2 100644 --- a/ppcls/arch/backbone/legendary_models/hrnet.py +++ b/ppcls/arch/backbone/legendary_models/hrnet.py @@ -397,7 +397,7 @@ class HRNet(TheseusLayer): stride=2, act='relu') - self.layer1 = self.bottleneck_blocks = nn.Sequential(*[ + self.layer1 = nn.Sequential(*[ BottleneckBlock( num_channels=64 if i == 0 else 256, num_filters=64,