update initialize_weights()

2020-04-14 15:58:32 -07:00 · 2020-04-14 15:58:32 -07:00 · a49ea80218
parent ac4c90c817
commit a49ea80218
1 changed files with 1 additions and 1 deletions
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@ -54,7 +54,7 @@ def initialize_weights(model):
    for m in model.modules():
        t = type(m)
        if t is nn.Conv2d:
-            nn.init.kaiming_normal_(m.weight, mode='fan_out', nonlinearity='relu')
+            pass  # nn.init.kaiming_normal_(m.weight, mode='fan_out', nonlinearity='relu')
        elif t is nn.BatchNorm2d:
            m.eps = 1e-4
            m.momentum = 0.03