ShaswatRobotics
/

world_model_test

ShaswatRobotics commited on Dec 29, 2025

Commit

c32c7e6

verified ·

1 Parent(s): 484653f

Update iris/atari/alien/config.json

Files changed (1) hide show

iris/atari/alien/config.json CHANGED Viewed

@@ -27,6 +27,8 @@
         "class_name": "WorldModel",
         "class_args": [
         {
           "tokens_per_block": 17,
           "max_blocks": 20,
           "attention": "causal",

         "class_name": "WorldModel",
         "class_args": [
         {
+          "vocab_size": 512,
+          "act_vocab_size": 18,
           "tokens_per_block": 17,
           "max_blocks": 20,
           "attention": "causal",