Add test and example config for DeepSetsPolicy

This commit is contained in:
Johannes Fischer
2021-07-19 18:38:50 +02:00
parent 0e6b1e102e
commit 49e32fe37b
4 changed files with 59 additions and 7 deletions

34
config/networks.json5 Normal file
View File

@@ -0,0 +1,34 @@
{
ego_state: {
input_dim: 5, // number of state vars
hidden_n: 1,
hidden_dim: 5,
output_dim: 5
},
deepsets: {
input_dim: 5, // number of relative state vars for others
phi: {
hidden_n: 1,
hidden_dim: 20,
},
latent_dim: 20,
rho: {
hidden_n: 1,
hidden_dim: 10,
},
output_dim: 10
},
path_encoder: {
input_dim: 40, // 2 * path length for (x,y) coordinates
hidden_n: 2,
hidden_dim: 20,
output_dim: 10,
},
head: {
input_dim: 0, // computed in policy constructor
hidden_n: 1,
hidden_dim: 50,
output_dim: 1, // number of outputs e.g. number of actions, or just one
final_activation: 'sigmoid',
}
}