huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change final value network activation to identity

Browse Source
This commit is contained in:
Johannes Fischer
2021-08-06 18:48:53 +02:00
parent 66bfba3986
commit 025c71767f
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
config/value_dice.json5
View File

@@ -65,7 +65,7 @@
hidden_n: 3, hidden_n: 3,
hidden_dim: 50, hidden_dim: 50,
output_dim: 1, // number of outputs e.g. number of actions, or just one output_dim: 1, // number of outputs e.g. number of actions, or just one
final_activation: 'sigmoid', final_activation: 'id',
}, },
}, },
policy_optim: { policy_optim: {

4
src/nets/util.py
View File

@@ -1,5 +1,5 @@
import torch import torch
from torch.nn import functional from torch.nn import functional, Identity
def parse_functional(functional_config): def parse_functional(functional_config):
if isinstance(functional_config, str): if isinstance(functional_config, str):
@@ -9,4 +9,6 @@ def parse_functional(functional_config):
return torch.sigmoid return torch.sigmoid
elif functional_config == 'softmax': elif functional_config == 'softmax':
return functional.softmax return functional.softmax
elif functional_config == 'id':
return Identity()
return None return None