Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update initialize_weights()

This commit is contained in:
Glenn Jocher 2020-04-14 15:58:32 -07:00
parent ac4c90c817
commit a49ea80218
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/torch_utils.py
View File

@ -54,7 +54,7 @@ def initialize_weights(model):
for m in model.modules(): for m in model.modules():
t = type(m) t = type(m)
if t is nn.Conv2d: 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: elif t is nn.BatchNorm2d:
m.eps = 1e-4 m.eps = 1e-4
m.momentum = 0.03 m.momentum = 0.03