mirror of
https://github.com/wassname/pytorch-soft-actor-critic.git
synced 2026-07-23 13:10:49 +08:00
Update model.py
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user