Update model.py

2026-08-18 12:20:57 +08:00 · 2019-01-02 22:24:04 +05:30
parent 717524b006
commit 0a7b9e51fb
1 changed files with 1 additions and 1 deletions
@@ -13,7 +13,7 @@ epsilon = 1e-6
 def weights_init(m):
    classname = m.__class__.__name__
    if classname.find('Linear') != -1:
-        torch.nn.init.xavier_normal_(m.weight)
+        torch.nn.init.xavier_uniform_(m.weight)


 class ValueNetwork(nn.Module):