Update model.py

This commit is contained in:
Pranjal Tandon
2018-10-10 00:23:23 +05:30
committed by GitHub
parent d222c9e83a
commit 2df9bb5954
+1 -1
View File
@@ -14,7 +14,7 @@ epsilon=1e-6
def weights_init_policy(m):
classname = m.__class__.__name__
if classname.find('Linear') != -1:
torch.nn.init.normal_(m.weight, mean=0, std=0.1)
torch.nn.init.normal_(m.weight, mean=0, std=0.0.075)
# Initialize QNetwork and Value Network weights
def weights_init_vf(m):