Upload SMAForSSL

config.json

@@ -0,0 +1,103 @@
+{
+  "_name_or_path": "/iris/u/jwxie/workspace/domain-agnostic-pretraining/examples/research_projects/domain-agnostic-pretraining/saved_models/image_pretrained/imagenet100_guided_rand_select_masking_recon_tiny_144_0.75_noise0.25-adamw_torch-lr1e-4-wd0.01-ws10000/checkpoint-390000",
+  "architectures": [
+    "SMAForSSL"
+  ],
+  "attention_dropout_prob": 0.0,
+  "cross_attention_widening_factor": 1,
+  "cross_eval_noising_args": null,
+  "cross_train_noising_args": [
+    [
+      "RandomlySelectedCrossAttentionMasking",
+      {
+        "exclude_seen_reconstruction": true,
+        "masking_ratio": 0.75,
+        "num_per_query": 144,
+        "select_initial_ratio": 1.0,
+        "varying_length": false
+      }
+    ]
+  ],
+  "decoder_attention_channels": 768,
+  "decoder_heads": 1,
+  "decoder_latent_channels": 768,
+  "decoder_type": "cross_attention",
+  "dense_use_bias": true,
+  "drop_path_rate": 0.0,
+  "embedded_channels": 196,
+  "encoder_cross_attention_channels": 196,
+  "encoder_type": "cross_attention",
+  "final_project": true,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "initializer_range": 0.02,
+  "input_channels": 3,
+  "input_type": "continuous",
+  "latent_channels": 768,
+  "layer_norm_eps": 1e-12,
+  "layernorm_eps": 1e-12,
+  "loss_fn": "mse",
+  "max_position_embeddings": 36864,
+  "model_type": "perceiver_sma",
+  "num_blocks": 1,
+  "num_cross_attention_heads": 1,
+  "num_discrete_tokens": 262,
+  "num_latents": 512,
+  "num_outputs": 1,
+  "num_self_attends_per_block": 8,
+  "num_self_attention_heads": 8,
+  "output_channels": 262,
+  "pe_initializer_range": 0.02,
+  "post_decoder_layers": null,
+  "project_after_concat": true,
+  "qk_channels": null,
+  "self_attention_widening_factor": 4,
+  "share_decoder_queries": true,
+  "share_embedding_weights": true,
+  "teacher_args": {
+    "auxiliary_loss_fn": "mse",
+    "auxiliary_loss_weight": 1.0,
+    "ema_args": {
+      "ema_decay_end": 0.0,
+      "ema_decay_start": 0.0
+    },
+    "eval_transform_args": [
+      [
+        "RandomlySelectedCrossAttentionMasking",
+        {
+          "exclude_seen_reconstruction": true,
+          "masking_ratio": 0.75,
+          "num_per_query": 144,
+          "select_initial_ratio": 1.0,
+          "varying_length": false
+        }
+      ]
+    ],
+    "mask_replace": null,
+    "norm_targets": true,
+    "num_layer_target_avg": null,
+    "reconstruction_decoder_args": {
+      "num_heads": 1,
+      "num_outputs": 36864,
+      "output_channels": 3,
+      "qk_channels": 196,
+      "query_num_channels": 196,
+      "share_decoder_queries": true,
+      "share_embedding_weights": true,
+      "use_query_residual": true,
+      "v_channels": 196
+    },
+    "reconstruction_loss_fn": "mse",
+    "reconstruction_loss_weight": 1.0,
+    "reconstruction_weighted_loss": false,
+    "target_normalization_fn": "layernorm",
+    "train_transform_args": null
+  },
+  "teacher_name": "ReconstructionTeacher",
+  "torch_dtype": "float32",
+  "transformers_version": "4.26.0.dev0",
+  "use_decoder": false,
+  "use_position_embeddings": true,
+  "use_query_residual": true,
+  "v_channels": 768
+}

pytorch_model.bin

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