[minor] fix a bug in weight transfer

mit-han-lab · Nov 10, 2022 · 3dd79dd · 3dd79dd
1 parent 04eca59
commit 3dd79dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/weight_transfer.py b/utils/weight_transfer.py
@@ -134,7 +134,7 @@ def transfer(m1, m2, input_index=None, output_index=None):
     elif isinstance(m1, SeparableConv2d):
         return transfer_SeparableConv2d(m1, m2, input_index, output_index)
     elif isinstance(m1, ResnetBlock):
-        return transfer_Conv2d(m1, m2, input_index, output_index)
+        return transfer_ResnetBlock(m1, m2, input_index, output_index)
     elif isinstance(m1, MobileResnetBlock):
         return transfer_MobileResnetBlock(m1, m2, input_index, output_index)
     elif isinstance(m1, MobileSPADEResnetBlock):