Add model

config.json

@@ -24,7 +24,7 @@
     ],
     "diversity_penalty": 0.0,
     "do_sample": false,
-    "drop_path_rate": 0.1,
+    "drop_path_rate": 0.3,
     "early_stopping": false,
     "embed_dim": 192,
     "encoder_no_repeat_ngram_size": 0,
@@ -2062,6 +2062,12 @@
     ],
     "num_layers": 4,
     "num_return_sequences": 1,
+    "out_features": [
+      "stage1",
+      "stage2",
+      "stage3",
+      "stage4"
+    ],
     "output_attentions": false,
     "output_hidden_states": false,
     "output_scores": false,
@@ -2077,6 +2083,13 @@
     "return_dict": true,
     "return_dict_in_generate": false,
     "sep_token_id": null,
+    "stage_names": [
+      "stem",
+      "stage1",
+      "stage2",
+      "stage3",
+      "stage4"
+    ],
     "suppress_tokens": null,
     "task_specific_params": null,
     "temperature": 1.0,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5e146d6c4fd780af86ad0c8e3fcdf88a62c696c060ce80439f7b4fb21218c0fd
-size 879533065
+oid sha256:e02211a290c7df91a89dc1065db701d83d85f170aaf1de793894d3fa7120371a
+size 879517453