learnable initial state options for RSSM

2023-04-29 07:54:03 +09:00
parent 1328ff1088
commit 0eb66997fb
4 changed files with 60 additions and 12 deletions
--- a/configs.yaml
+++ b/configs.yaml
@@ -63,6 +63,7 @@ defaults:
  weight_decay: 0.0
  unimix_ratio: 0.01
  action_unimix_ratio: 0.01
+  initial: 'learned'

  # Training
  batch_size: 16