ShaswatRobotics commited on
Commit
dd71097
·
verified ·
1 Parent(s): 92ea0b7

Update delta-iris/crafter/config.json

Browse files
Files changed (1) hide show
  1. delta-iris/crafter/config.json +3 -2
delta-iris/crafter/config.json CHANGED
@@ -18,7 +18,8 @@
18
  "weights_path": "last.pt",
19
  "class_path": "../src/world_model.py",
20
  "class_name": "WorldModel",
21
- "class_kwargs":{
 
22
  "latent_vocab_size": 1024,
23
  "num_actions": 17,
24
  "image_channels": 3,
@@ -59,7 +60,7 @@
59
  "weights_path": "last.pt",
60
  "class_path": "../src/tokenizer.py",
61
  "class_name": "Tokenizer",
62
- "class_kwargs":{
63
  "image_channels": 3,
64
  "image_size": 64,
65
  "num_actions": 17,
 
18
  "weights_path": "last.pt",
19
  "class_path": "../src/world_model.py",
20
  "class_name": "WorldModel",
21
+ "class_args":
22
+ {
23
  "latent_vocab_size": 1024,
24
  "num_actions": 17,
25
  "image_channels": 3,
 
60
  "weights_path": "last.pt",
61
  "class_path": "../src/tokenizer.py",
62
  "class_name": "Tokenizer",
63
+ "class_args":{
64
  "image_channels": 3,
65
  "image_size": 64,
66
  "num_actions": 17,