estebancarlin commited on
Commit
93d1644
·
verified ·
1 Parent(s): e6cee75

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -0
config.json CHANGED
@@ -22,6 +22,8 @@
22
  "fusion_hidden_size": 128,
23
  "fusion_num_heads": 4,
24
  "fusion_num_layers": 2,
 
 
25
  "memory_alpha": 0.2,
26
  "direct_writing": true,
27
  "memory_compression": true,
 
22
  "fusion_hidden_size": 128,
23
  "fusion_num_heads": 4,
24
  "fusion_num_layers": 2,
25
+ "memory_size": 32,
26
+ "episode_dim": 128,
27
  "memory_alpha": 0.2,
28
  "direct_writing": true,
29
  "memory_compression": true,