Push model using huggingface_hub.

README.md

@@ -0,0 +1,9 @@
+---
+tags:
+- model_hub_mixin
+- pytorch_model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Library: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,41 @@
+{
+  "action_space": null,
+  "policy_kwargs": {
+    "active_reward_monitors": {
+      "env_reward": 1
+    },
+    "attention_heads": 24,
+    "attention_mask_style": "clipped_causal",
+    "attention_memory_size": 256,
+    "diff_mlp_embedding": false,
+    "hidsize": 3072,
+    "img_shape": [
+      128,
+      128,
+      3
+    ],
+    "impala_chans": [
+      16,
+      32,
+      32
+    ],
+    "impala_kwargs": {
+      "post_pool_groups": 1
+    },
+    "impala_width": 12,
+    "init_norm_kwargs": {
+      "batch_norm": false,
+      "group_norm_groups": 1
+    },
+    "n_recurrence_layers": 4,
+    "only_img_input": true,
+    "pointwise_ratio": 4,
+    "pointwise_use_activation": false,
+    "recurrence_is_residual": true,
+    "recurrence_type": "transformer",
+    "timesteps": 128,
+    "use_pointwise_layer": true,
+    "use_pre_lstm_ln": false
+  },
+  "temperature": 3
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8be213ef06de248e94b0b5c65853118faa923c8feeee5365f6f71e07cf172129
+size 2130033464