diff --git a/pose/models/hourglass.py b/pose/models/hourglass.py
index a1b84af..25c7ec6 100644
--- a/pose/models/hourglass.py
+++ b/pose/models/hourglass.py
@@ -145,7 +145,7 @@ def _make_residual(self, block, planes, blocks, stride=1):
         return nn.Sequential(*layers)
 
     def _make_fc(self, inplanes, outplanes):
-        bn = nn.BatchNorm2d(inplanes)
+        bn = nn.BatchNorm2d(outplanes)
         conv = nn.Conv2d(inplanes, outplanes, kernel_size=1, bias=True)
         return nn.Sequential(
                 conv,
diff --git a/pose/models/hourglass_gn.py b/pose/models/hourglass_gn.py
index a780ac5..2a9ded2 100644
--- a/pose/models/hourglass_gn.py
+++ b/pose/models/hourglass_gn.py
@@ -148,7 +148,7 @@ def _make_residual(self, block, planes, blocks, stride=1):
         return nn.Sequential(*layers)
 
     def _make_fc(self, inplanes, outplanes):
-        bn = nn.GroupNorm(gn, inplanes)
+        bn = nn.GroupNorm(gn, outplanes)
         conv = nn.Conv2d(inplanes, outplanes, kernel_size=1, bias=True)
         return nn.Sequential(
                 conv,