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