From 2df9bb595451bbecc22cbc18a02439db3ac7ce91 Mon Sep 17 00:00:00 2001 From: Pranjal Tandon Date: Wed, 10 Oct 2018 00:23:23 +0530 Subject: [PATCH] Update model.py --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index cee5fd5..bbf7c4c 100644 --- a/model.py +++ b/model.py @@ -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):