Update delta-iris/crafter/config.json
Browse files
delta-iris/crafter/config.json
CHANGED
|
@@ -17,7 +17,35 @@
|
|
| 17 |
"source": {
|
| 18 |
"weights_path": "last.pt",
|
| 19 |
"class_path": "../src/world_model.py",
|
| 20 |
-
"class_name": "WorldModel"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
},
|
| 22 |
"signature": {
|
| 23 |
"call_mode": "auto"
|
|
@@ -30,7 +58,38 @@
|
|
| 30 |
"source": {
|
| 31 |
"weights_path": "last.pt",
|
| 32 |
"class_path": "../src/tokenizer.py",
|
| 33 |
-
"class_name": "Tokenizer"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
},
|
| 35 |
"signature": {
|
| 36 |
"call_mode": "auto"
|
|
|
|
| 17 |
"source": {
|
| 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,
|
| 25 |
+
"image_size": 64,
|
| 26 |
+
"latents_weight": 1.0,
|
| 27 |
+
"rewards_weight": 1.0,
|
| 28 |
+
"ends_weight": 1.0,
|
| 29 |
+
"two_hot_rews": true,
|
| 30 |
+
"transformer_config": {
|
| 31 |
+
"tokens_per_block": 6,
|
| 32 |
+
"max_blocks": 21,
|
| 33 |
+
"num_layers": 3,
|
| 34 |
+
"num_heads": 8,
|
| 35 |
+
"embed_dim": 512,
|
| 36 |
+
"attention": "causal",
|
| 37 |
+
"embed_pdrop": 0.0,
|
| 38 |
+
"resid_pdrop": 0.0,
|
| 39 |
+
"attn_pdrop": 0.0,
|
| 40 |
+
},
|
| 41 |
+
"frame_cnn_config": {
|
| 42 |
+
"image_channels": 3,
|
| 43 |
+
"latent_dim": 8,
|
| 44 |
+
"num_channels": 32,
|
| 45 |
+
"mult": [1, 1, 2, 2, 4],
|
| 46 |
+
"down": [1, 0, 1, 1, 0],
|
| 47 |
+
}
|
| 48 |
+
}
|
| 49 |
},
|
| 50 |
"signature": {
|
| 51 |
"call_mode": "auto"
|
|
|
|
| 58 |
"source": {
|
| 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,
|
| 66 |
+
"num_tokens": 4,
|
| 67 |
+
"decoder_act_channels": 4,
|
| 68 |
+
"codebook_size": 1024,
|
| 69 |
+
"codebook_dim": 64,
|
| 70 |
+
"max_codebook_updates_with_revival": 0,
|
| 71 |
+
"encoder_config": {
|
| 72 |
+
"image_channels": 7,
|
| 73 |
+
"latent_dim": 64,
|
| 74 |
+
"num_channels": 64,
|
| 75 |
+
"mult": [1, 1, 2, 2, 4],
|
| 76 |
+
"down": [1, 0, 1, 1, 0]
|
| 77 |
+
},
|
| 78 |
+
"decoder_config": {
|
| 79 |
+
"image_channels": 3,
|
| 80 |
+
"latent_dim": 76,
|
| 81 |
+
"num_channels": 64,
|
| 82 |
+
"mult": [1, 1, 2, 2, 4],
|
| 83 |
+
"down": [1, 0, 1, 1, 0],
|
| 84 |
+
},
|
| 85 |
+
"frame_cnn_config": {
|
| 86 |
+
"image_channels": 3,
|
| 87 |
+
"latent_dim": 8,
|
| 88 |
+
"num_channels": 64,
|
| 89 |
+
"mult": [1, 1, 2, 2, 4],
|
| 90 |
+
"down": [1, 0, 1, 1, 0]
|
| 91 |
+
}
|
| 92 |
+
}
|
| 93 |
},
|
| 94 |
"signature": {
|
| 95 |
"call_mode": "auto"
|