diff options
| author | Taesung Park <taesung89@gmail.com> | 2017-12-01 22:13:51 -0800 |
|---|---|---|
| committer | Taesung Park <taesung89@gmail.com> | 2017-12-01 22:13:51 -0800 |
| commit | 7bda734dd7f3466d5d55afe80b97542b1b12bdb5 (patch) | |
| tree | 82cfdacd331189c4367745a510f09aef833c77a6 /models/networks.py | |
| parent | 1615932f9180a7a9df92f33fbb8749aec432d3d9 (diff) | |
changed the gain of xavier initialization from 1 to 0.02. implemented serial_batches option in unaligned dataset
Diffstat (limited to 'models/networks.py')
| -rw-r--r-- | models/networks.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/models/networks.py b/models/networks.py index e6e0a87..3c54138 100644 --- a/models/networks.py +++ b/models/networks.py @@ -26,9 +26,9 @@ def weights_init_xavier(m): classname = m.__class__.__name__ # print(classname) if classname.find('Conv') != -1: - init.xavier_normal(m.weight.data, gain=1) + init.xavier_normal(m.weight.data, gain=0.02) elif classname.find('Linear') != -1: - init.xavier_normal(m.weight.data, gain=1) + init.xavier_normal(m.weight.data, gain=0.02) elif classname.find('BatchNorm2d') != -1: init.normal(m.weight.data, 1.0, 0.02) init.constant(m.bias.data, 0.0) |