Hugging Face's logo

kanixwang
/
eth-setfit-payment-model

Sentence Similarity
sentence-transformers
PyTorch
Transformers
mpnet
feature-extraction
Inference Endpoints
Model card Files Community
kanixwang commited on
Commit
26040f5
·
1 Parent(s): 10ed2dc

Add SetFit 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 26310 with parameters:
89
  ```
90
  {'batch_size': 8, '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": 26310,
110
- "warmup_steps": 2631,
111
  "weight_decay": 0.01
112
  }
113
  ```
 
85
 
86
  **DataLoader**:
87
 
88
+ `torch.utils.data.dataloader.DataLoader` of length 26915 with parameters:
89
  ```
90
  {'batch_size': 8, '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": 26915,
110
+ "warmup_steps": 2692,
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:f8f4840f27e39a16915082aeafc8d15436f8a88026a0254716b5b001fca66d8f
3
  size 68591
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c67268f5110d6e2d55569c451f18c6d013c47f0a92a3ae954c7b600d477c653
3
  size 68591
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b305885f8983a94c7807d0da63a1444d2375a51ec93843231ee6c4ede98c0577
3
  size 438014769
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58f14ae6a5df496e8a80016e30116794dfa3c917ee8c6bd00690eff5a632d4f8
3
  size 438014769