Add nn.SiLU inplace in attempt_load() (#1940)

* sleep 3 sec to load data . * Update train.py * Add nn.SiLU inplace in attempt_load() Co-authored-by: wangliang <[email protected]> Co-authored-by: Glenn Jocher <[email protected]>
ultralytics · Jan 15, 2021 · 03ebe6e · 03ebe6e
1 parent 777032e
commit 03ebe6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/experimental.py b/models/experimental.py
@@ -119,7 +119,7 @@ def attempt_load(weights, map_location=None):
 
     # Compatibility updates
     for m in model.modules():
-        if type(m) in [nn.Hardswish, nn.LeakyReLU, nn.ReLU, nn.ReLU6]:
+        if type(m) in [nn.Hardswish, nn.LeakyReLU, nn.ReLU, nn.ReLU6, nn.SiLU]:
             m.inplace = True  # pytorch 1.7.0 compatibility
         elif type(m) is Conv:
             m._non_persistent_buffers_set = set()  # pytorch 1.6.0 compatibility