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

Add debug output and original implementation

Browse Source
This commit is contained in:
Johannes Fischer
2021-09-08 13:52:21 +02:00
parent 1fd0a71646
commit 26cde18be3
3 changed files with 149 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
config/value_dice.json5
View File

@@ -9,12 +9,12 @@
deepsets: {
input_dim: 6, // number of relative state vars for others
phi: {
hidden_n: 2,
hidden_n: 1,
hidden_dim: 20,
},
latent_dim: 20,
rho: {
hidden_n: 2,
hidden_n: 1,
hidden_dim: 10,
},
output_dim: 10
@@ -43,12 +43,12 @@
deepsets: {
input_dim: 6, // number of relative state vars for others
phi: {
hidden_n: 2,
hidden_n: 1,
hidden_dim: 20,
},
latent_dim: 20,
rho: {
hidden_n: 2,
hidden_n: 1,
hidden_dim: 10,
},
output_dim: 10
@@ -70,13 +70,13 @@
},
policy_optim: {
optimizer: 'adam',
lr: 1e-3,
weight_decay: 0.1,
lr: 1e-0,
weight_decay: 0.01,
},
value_optim: {
optimizer: 'adam',
lr: 1e-3,
weight_decay: 0.1,
lr: 1e-6,
weight_decay: 0.01,
},
train_epochs: 200,
train_batch_size: 32,