config

config.json

@@ -0,0 +1,42 @@
+{
+    "model_type": "dinosaur",
+    "initializer": {
+      "name": "RandomInit",
+      "n_slots": 7,
+      "dim": 512
+    },
+    "encoder": {
+      "backbone": {
+        "name": "TimmExtractor",
+        "model": "vit_large_patch14_reg4_dinov2.lvd142m",
+        "features": "vit_block12",
+        "frozen": true,
+        "pretrained": true,
+        "model_kwargs": {
+          "dynamic_img_size": true
+        }
+      },
+      "output_transform": {
+        "name": "networks.two_layer_mlp",
+        "inp_dim": 1024,
+        "outp_dim": 512,
+        "hidden_dim": 2048,
+        "layer_norm": true
+      }
+    },
+    "grouper": {
+      "name": "SlotAttention",
+      "inp_dim": 512,
+      "slot_dim": 512,
+      "n_iters": 3,
+      "use_mlp": true
+    },
+    "decoder": {
+      "name": "MLPDecoder",
+      "inp_dim": 512,
+      "outp_dim": 1024,
+      "hidden_dims": [2048, 2048, 2048],
+      "n_patches": 676
+    },
+    "aux_outputs": false
+  }