From a49ea80218f21e1c5d3b9340a75bb3d2deb106a3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 14 Apr 2020 15:58:32 -0700 Subject: [PATCH] update initialize_weights() --- utils/torch_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index e5286cd0..c6690924 100644 --- 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