Update delta-iris/atari/boxing/config.json
Browse files
delta-iris/atari/boxing/config.json
CHANGED
|
@@ -61,6 +61,7 @@
|
|
| 61 |
[
|
| 62 |
{
|
| 63 |
"name": "act_emb",
|
|
|
|
| 64 |
"signature":
|
| 65 |
{
|
| 66 |
"inputs": ["act"],
|
|
@@ -69,6 +70,7 @@
|
|
| 69 |
},
|
| 70 |
{
|
| 71 |
"name": "latents_emb",
|
|
|
|
| 72 |
"signature":
|
| 73 |
{
|
| 74 |
"inputs": ["latent_tokens"],
|
|
@@ -77,6 +79,7 @@
|
|
| 77 |
},
|
| 78 |
{
|
| 79 |
"name": "transformer",
|
|
|
|
| 80 |
"signature":
|
| 81 |
{
|
| 82 |
"call_mode": "auto"
|
|
@@ -84,6 +87,7 @@
|
|
| 84 |
},
|
| 85 |
{
|
| 86 |
"name": "frame_cnn",
|
|
|
|
| 87 |
"signature":
|
| 88 |
{
|
| 89 |
"inputs": ["obs"],
|
|
@@ -140,6 +144,7 @@
|
|
| 140 |
[
|
| 141 |
{
|
| 142 |
"name": "decode",
|
|
|
|
| 143 |
"signature":
|
| 144 |
{
|
| 145 |
"inputs": ["obs", "act", "q", "should_clamp"],
|
|
|
|
| 61 |
[
|
| 62 |
{
|
| 63 |
"name": "act_emb",
|
| 64 |
+
"sub_model_name": "act_emb",
|
| 65 |
"signature":
|
| 66 |
{
|
| 67 |
"inputs": ["act"],
|
|
|
|
| 70 |
},
|
| 71 |
{
|
| 72 |
"name": "latents_emb",
|
| 73 |
+
"sub_model_name": "latents_emb",
|
| 74 |
"signature":
|
| 75 |
{
|
| 76 |
"inputs": ["latent_tokens"],
|
|
|
|
| 79 |
},
|
| 80 |
{
|
| 81 |
"name": "transformer",
|
| 82 |
+
"sub_model_name": "transformer",
|
| 83 |
"signature":
|
| 84 |
{
|
| 85 |
"call_mode": "auto"
|
|
|
|
| 87 |
},
|
| 88 |
{
|
| 89 |
"name": "frame_cnn",
|
| 90 |
+
"sub_model_name": "frame_cnn",
|
| 91 |
"signature":
|
| 92 |
{
|
| 93 |
"inputs": ["obs"],
|
|
|
|
| 144 |
[
|
| 145 |
{
|
| 146 |
"name": "decode",
|
| 147 |
+
"sub_model_name": "decode",
|
| 148 |
"signature":
|
| 149 |
{
|
| 150 |
"inputs": ["obs", "act", "q", "should_clamp"],
|