initial commit

config.json

@@ -0,0 +1,29 @@
+{
+  "_name_or_path": "serialization_dir/distilelectra.pth",
+  "architectures": [
+    "ElectraForPreTraining"
+  ],
+  "attention_probs_dropout_prob": 0.1,
+  "classifier_dropout": null,
+  "embedding_size": 768,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 512,
+  "model_type": "electra",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 6,
+  "output_past": true,
+  "pad_token_id": 0,
+  "position_embedding_type": "absolute",
+  "summary_activation": "gelu",
+  "summary_last_dropout": 0.1,
+  "summary_type": "first",
+  "summary_use_proj": true,
+  "transformers_version": "4.11.3",
+  "type_vocab_size": 2,
+  "vocab_size": 30522
+}

git_log.json

@@ -0,0 +1,5 @@
+{
+    "repo_id": "<git.repo.base.Repo '/home/jovyan/distil_tmp/distillation/.git'>",
+    "repo_sha": "de6dcaeed8e16839757d0c5a3a5f7e60f0b454e6",
+    "repo_branch": "master"
+}

parameters.json

@@ -0,0 +1,51 @@
+{
+    "force": true,
+    "dump_path": "./serialization_dir/distilbert_mse",
+    "data_file": "data/binarized_text.electra.pickle",
+    "student_type": "electra",
+    "student_config": "training_configs/distilelectra.json",
+    "student_pretrained_weights": "serialization_dir/distilelectra.pth",
+    "teacher_type": "electra",
+    "teacher_name": "google/electra-base-discriminator",
+    "temperature": 2.0,
+    "alpha_ce": 5.0,
+    "alpha_mlm": 2.0,
+    "alpha_clm": 0.0,
+    "alpha_mse": 1.0,
+    "alpha_cos": 1.0,
+    "mlm": true,
+    "mlm_mask_prop": 0.15,
+    "word_mask": 0.8,
+    "word_keep": 0.1,
+    "word_rand": 0.1,
+    "mlm_smoothing": 0.7,
+    "token_counts": "data/token_counts.electra.pickle",
+    "restrict_ce_to_mask": false,
+    "freeze_pos_embs": true,
+    "freeze_token_type_embds": false,
+    "n_epoch": 50,
+    "batch_size": 5,
+    "group_by_size": true,
+    "gradient_accumulation_steps": 50,
+    "warmup_prop": 0.05,
+    "weight_decay": 0.0,
+    "learning_rate": 0.0005,
+    "adam_epsilon": 1e-06,
+    "max_grad_norm": 5.0,
+    "initializer_range": 0.02,
+    "fp16": false,
+    "fp16_opt_level": "O1",
+    "n_gpu": 4,
+    "local_rank": 0,
+    "seed": 56,
+    "log_interval": 500,
+    "checkpoint_interval": 4000,
+    "world_size": 4,
+    "n_gpu_per_node": 4,
+    "global_rank": 0,
+    "n_nodes": 1,
+    "node_id": 0,
+    "multi_gpu": true,
+    "is_master": true,
+    "multi_node": false
+}

pytorch_model.bin

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