Push model using huggingface_hub.

README.md

@@ -1,15 +1,10 @@
 ---
-language: en
-license: mit
 tags:
-- pytorch
-- pertTF
+- model_hub_mixin
+- pytorch_model_hub_mixin
 ---
 
-# pertTF perturbation model (tiny, mvc_only)
-
-This is model files for pertTF (tiny) trained for perturbation prediction using only mvc loss
-
-## Usage
-```python
-model = HFPerturbationTFModel.from_pretrained('weililab/pert_split_1_5k_mvc_only')
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: [More Information Needed]
+- Paper: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -19,7 +19,7 @@
   "nlayers": 2,
   "nlayers_cls": 3,
   "nlayers_pert": 4,
-  "ntoken": 5056,
+  "ntoken": 5057,
   "num_batch_labels": null,
   "pad_token": "<pad>",
   "pad_value": -2,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6de165bfa12df6be342e1dd452c820e7f1c2e7daec67dbec329af372418b5993
-size 3361088
+oid sha256:f22107893737d8677143ea2d6505d85ae0a46b1d5597ea16e55f036637fa430c
+size 3361216

training_config.json

@@ -1,6 +1,6 @@
 {
   "seed": 42,
-  "dataset_name": "split_1_5k",
+  "dataset_name": "split_TRAIN_5k",
   "do_train": true,
   "load_model": null,
   "GEPC": true,
@@ -45,11 +45,11 @@
   "cls_value": -3,
   "cls_token": "<cls>",
   "sampling_mode": "hvg",
-  "non_hvg_size": 50,
+  "non_hvg_size": 1000,
   "fix_nonzero_prop": true,
   "nonzero_prop": 0.9,
   "hvg_col": "highly_variable",
   "mvc_masked_train": true,
-  "max_seq_len": 3051,
+  "max_seq_len": 4001,
   "pad_value": -2
 }