96abhishekarora commited on
Commit
21d7f88
1 Parent(s): 1a5f5d0

Modified validation and training for linktransformer model

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. README.md +3 -3
  3. pytorch_model.bin +1 -1
.gitattributes CHANGED
@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
 
 
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
37
+ .git/lfs/objects/a9/a3/a9a34a666447dbbd0b4be8ba7865bf6748abd81f63dcb52668c47681cbe29c03 filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -97,7 +97,7 @@ The model was trained with the parameters:
97
 
98
  **DataLoader**:
99
 
100
- `torch.utils.data.dataloader.DataLoader` of length 281 with parameters:
101
  ```
102
  {'batch_size': 64, 'sampler': 'torch.utils.data.dataloader._InfiniteConstantSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
103
  ```
@@ -110,7 +110,7 @@ Parameters of the fit()-Method:
110
  ```
111
  {
112
  "epochs": 100,
113
- "evaluation_steps": 2810,
114
  "evaluator": "sentence_transformers.evaluation.SequentialEvaluator.SequentialEvaluator",
115
  "max_grad_norm": 1,
116
  "optimizer_class": "<class 'torch.optim.adamw.AdamW'>",
@@ -119,7 +119,7 @@ Parameters of the fit()-Method:
119
  },
120
  "scheduler": "WarmupLinear",
121
  "steps_per_epoch": null,
122
- "warmup_steps": 28100,
123
  "weight_decay": 0.01
124
  }
125
  ```
 
97
 
98
  **DataLoader**:
99
 
100
+ `torch.utils.data.dataloader.DataLoader` of length 279 with parameters:
101
  ```
102
  {'batch_size': 64, 'sampler': 'torch.utils.data.dataloader._InfiniteConstantSampler', 'batch_sampler': 'torch.utils.data.sampler.BatchSampler'}
103
  ```
 
110
  ```
111
  {
112
  "epochs": 100,
113
+ "evaluation_steps": 2790,
114
  "evaluator": "sentence_transformers.evaluation.SequentialEvaluator.SequentialEvaluator",
115
  "max_grad_norm": 1,
116
  "optimizer_class": "<class 'torch.optim.adamw.AdamW'>",
 
119
  },
120
  "scheduler": "WarmupLinear",
121
  "steps_per_epoch": null,
122
+ "warmup_steps": 27900,
123
  "weight_decay": 0.01
124
  }
125
  ```
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a9a34a666447dbbd0b4be8ba7865bf6748abd81f63dcb52668c47681cbe29c03
3
  size 1346772393
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:488aa4746aae10652265c4f48d8ddaaf8e302e6fbd4a5db9da494309eaa8eeac
3
  size 1346772393