update config

config.json

@@ -1,10 +1,10 @@
 {
-  "_name_or_path": "experiments/wandb/run-20240605_134328-5wbd9pb9/files/huggingface_checkpoint",
   "add_extra_token": true,
   "architectures": [
     "SetEncoderElectraModel"
   ],
   "attention_probs_dropout_prob": 0.1,
+  "backbone_model_type": "electra",
   "classifier_dropout": null,
   "depth": 100,
   "doc_length": 256,
@@ -16,15 +16,14 @@
   "intermediate_size": 4096,
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
-  "model_type": "set-encoder-electra",
+  "model_type": "set-encoder",
   "num_attention_heads": 16,
   "num_hidden_layers": 24,
-  "other_sequence_embedding": false,
   "pad_token_id": 0,
+  "pooling_strategy": "first",
   "position_embedding_type": "absolute",
   "query_length": 32,
   "sample_missing_docs": true,
-  "save_step": 939,
   "summary_activation": "gelu",
   "summary_last_dropout": 0.1,
   "summary_type": "first",