refactor: Fixed Mish integration

frgfm · Nov 6, 2021 · a8b4fdb · a8b4fdb
1 parent f8e3181
commit a8b4fdb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/holocron/models/detection/yolov4.py b/holocron/models/detection/yolov4.py
@@ -456,7 +456,7 @@ def __init__(
             backbone_norm_layer = norm_layer
 
         # backbone
-        self.backbone = DarknetBodyV4(layout, in_channels, stem_channels, 3, Mish(),
+        self.backbone = DarknetBodyV4(layout, in_channels, stem_channels, 3, nn.Mish(inplace=True),
                                       backbone_norm_layer, drop_layer, conv_layer)
         # neck
         self.neck = Neck([1024, 512, 256], act_layer, norm_layer, drop_layer, conv_layer)