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

changing default network, making deepsets network choose output dimension appropriately, making bc config to be called by ray

Browse Source
This commit is contained in:
Arec
2021-07-26 07:52:11 -07:00
parent 69359b5af3
commit d9436daaba
4 changed files with 42 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
config/networks.json5
View File

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