ShaswatRobotics
/

world_model_test

ShaswatRobotics commited on 23 days ago

Commit

7592624

verified ·

1 Parent(s): b4a268f

Update delta-iris/atari/boxing/config.json

Files changed (1) hide show

delta-iris/atari/boxing/config.json CHANGED Viewed

@@ -24,7 +24,7 @@
       "format": "state_dict",
       "source": {
         "weights_path": "world_model.pt",
-        "class_path": "../src/world_model.py",
         "class_name": "WorldModel",
         "class_args": [
         {
@@ -101,7 +101,7 @@
       "format": "state_dict",
       "source": {
         "weights_path": "tokenizer.pt",
-        "class_path": "../src/tokenizer.py",
         "class_name": "Tokenizer",
         "class_args": [{
           "image_channels": 3,

       "format": "state_dict",
       "source": {
         "weights_path": "world_model.pt",
+        "class_path": "../../src/world_model.py",
         "class_name": "WorldModel",
         "class_args": [
         {
       "format": "state_dict",
       "source": {
         "weights_path": "tokenizer.pt",
+        "class_path": "../../src/tokenizer.py",
         "class_name": "Tokenizer",
         "class_args": [{
           "image_channels": 3,