diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-11-27 22:05:28 -0800 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-11-27 22:05:28 -0800 |
| commit | 1171d57b5db2d997a5ba3739e680af261f3a2e8a (patch) | |
| tree | 6d64c14df246a0fd2226b542a0340f0eac7937fe /models/networks.py | |
| parent | dcc954f025d62d0bf0a8ae3e071ccbffa5b4debe (diff) | |
fixed a bug in initialization of weights
Diffstat (limited to 'models/networks.py')
| -rw-r--r-- | models/networks.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/models/networks.py b/models/networks.py index ec6573b..e6e0a87 100644 --- a/models/networks.py +++ b/models/networks.py @@ -14,11 +14,11 @@ def weights_init_normal(m): classname = m.__class__.__name__ # print(classname) if classname.find('Conv') != -1: - init.uniform(m.weight.data, 0.0, 0.02) + init.normal(m.weight.data, 0.0, 0.02) elif classname.find('Linear') != -1: - init.uniform(m.weight.data, 0.0, 0.02) + init.normal(m.weight.data, 0.0, 0.02) elif classname.find('BatchNorm2d') != -1: - init.uniform(m.weight.data, 1.0, 0.02) + init.normal(m.weight.data, 1.0, 0.02) init.constant(m.bias.data, 0.0) @@ -30,7 +30,7 @@ def weights_init_xavier(m): elif classname.find('Linear') != -1: init.xavier_normal(m.weight.data, gain=1) elif classname.find('BatchNorm2d') != -1: - init.uniform(m.weight.data, 1.0, 0.02) + init.normal(m.weight.data, 1.0, 0.02) init.constant(m.bias.data, 0.0) @@ -42,7 +42,7 @@ def weights_init_kaiming(m): elif classname.find('Linear') != -1: init.kaiming_normal(m.weight.data, a=0, mode='fan_in') elif classname.find('BatchNorm2d') != -1: - init.uniform(m.weight.data, 1.0, 0.02) + init.normal(m.weight.data, 1.0, 0.02) init.constant(m.bias.data, 0.0) @@ -54,7 +54,7 @@ def weights_init_orthogonal(m): elif classname.find('Linear') != -1: init.orthogonal(m.weight.data, gain=1) elif classname.find('BatchNorm2d') != -1: - init.uniform(m.weight.data, 1.0, 0.02) + init.normal(m.weight.data, 1.0, 0.02) init.constant(m.bias.data, 0.0) |