Hugging Face's logo

futuredatascience
/
action-classifier-v0

Sentence Similarity
sentence-transformers
PyTorch
Transformers
mpnet
feature-extraction
Inference Endpoints
Model card Files Community
dpeletz commited on
Commit
ce05ae8
•
1 Parent(s): a346aea

Add model

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. model_head.pkl +1 -1
  3. pytorch_model.bin +1 -1
README.md CHANGED
@@ -85,7 +85,7 @@ The model was trained with the parameters:
85
 
86
  **DataLoader**:
87
 
88
- `torch.utils.data.dataloader.DataLoader` of length 50 with parameters:
89
  ```
90
  {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
91
  ```
@@ -106,8 +106,8 @@ Parameters of the fit()-Method:
106
  "lr": 2e-05
107
  },
108
  "scheduler": "WarmupLinear",
109
- "steps_per_epoch": 500,
110
- "warmup_steps": 50,
111
  "weight_decay": 0.01
112
  }
113
  ```
 
85
 
86
  **DataLoader**:
87
 
88
+ `torch.utils.data.dataloader.DataLoader` of length 60 with parameters:
89
  ```
90
  {'batch_size': 16, 'sampler': 'torch.utils.data.sampler.RandomSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
91
  ```
 
106
  "lr": 2e-05
107
  },
108
  "scheduler": "WarmupLinear",
109
+ "steps_per_epoch": 600,
110
+ "warmup_steps": 60,
111
  "weight_decay": 0.01
112
  }
113
  ```
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1b08582edb7d3a8287ccec7d5f40db7c3c4d59a331d8ebf147eedd7d62e7ce5f
3
  size 7118
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f663c8a56cd7807761c2e2bea9166709037a8f8fd36072eb7e9449e88c2c7a6e
3
  size 7118
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:69b362263646ef609da860a73bf1d5ba837a937d633dad78af9b51bcde6fcaca
3
  size 438014769
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f94d260eb1be9387720caf4b7f9eafdcd51dd2efe999ed5370ba1646a90f2dd2
3
  size 438014769