Instructions to use circlemachinelearning/bart-email-multitask with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use circlemachinelearning/bart-email-multitask with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("circlemachinelearning/bart-email-multitask") model = AutoModelForSeq2SeqLM.from_pretrained("circlemachinelearning/bart-email-multitask", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Training in progress, step 1500, checkpoint
Browse files
last-checkpoint/model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1625422896
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5a278fb208cf82d5c5728afb7c555b5b65b6f6a012bb67285a69e4eb073254f6
|
| 3 |
size 1625422896
|
last-checkpoint/optimizer.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 3250759392
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:709293e8f8d89d46e62d8dcffb82aec4b6a3854c0c787c3905239ef720804708
|
| 3 |
size 3250759392
|
last-checkpoint/rng_state.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 14645
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5420c023313a95f831e40048e5f377d6fb45506c03fe462bce53dbc170b39da5
|
| 3 |
size 14645
|
last-checkpoint/scaler.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1383
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:016a72d1342888aeb454c6fd7e33114f8fe7d8d6d4c795719a5831d14388e2ea
|
| 3 |
size 1383
|
last-checkpoint/scheduler.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1465
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b44fb2a9542e90532b93ebac4a5e8666f98d6d7698a2bf80999186ee0ee3983
|
| 3 |
size 1465
|
last-checkpoint/trainer_state.json
CHANGED
|
@@ -2,9 +2,9 @@
|
|
| 2 |
"best_global_step": 1000,
|
| 3 |
"best_metric": 50.66,
|
| 4 |
"best_model_checkpoint": "./bart-email-multitask-updated/checkpoint-1000",
|
| 5 |
-
"epoch": 0.
|
| 6 |
"eval_steps": 500,
|
| 7 |
-
"global_step":
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
@@ -174,6 +174,89 @@
|
|
| 174 |
"eval_samples_per_second": 1.092,
|
| 175 |
"eval_steps_per_second": 0.273,
|
| 176 |
"step": 1000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 177 |
}
|
| 178 |
],
|
| 179 |
"logging_steps": 50,
|
|
@@ -188,7 +271,7 @@
|
|
| 188 |
"early_stopping_threshold": 0.0
|
| 189 |
},
|
| 190 |
"attributes": {
|
| 191 |
-
"early_stopping_patience_counter":
|
| 192 |
}
|
| 193 |
},
|
| 194 |
"TrainerControl": {
|
|
@@ -202,7 +285,7 @@
|
|
| 202 |
"attributes": {}
|
| 203 |
}
|
| 204 |
},
|
| 205 |
-
"total_flos":
|
| 206 |
"train_batch_size": 4,
|
| 207 |
"trial_name": null,
|
| 208 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": 1000,
|
| 3 |
"best_metric": 50.66,
|
| 4 |
"best_model_checkpoint": "./bart-email-multitask-updated/checkpoint-1000",
|
| 5 |
+
"epoch": 0.517821696729093,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 1500,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 174 |
"eval_samples_per_second": 1.092,
|
| 175 |
"eval_steps_per_second": 0.273,
|
| 176 |
"step": 1000
|
| 177 |
+
},
|
| 178 |
+
{
|
| 179 |
+
"epoch": 0.3624751877103651,
|
| 180 |
+
"grad_norm": 4.5150227546691895,
|
| 181 |
+
"learning_rate": 4.241151233464427e-06,
|
| 182 |
+
"loss": 0.4227,
|
| 183 |
+
"step": 1050
|
| 184 |
+
},
|
| 185 |
+
{
|
| 186 |
+
"epoch": 0.3797359109346682,
|
| 187 |
+
"grad_norm": 2.6075778007507324,
|
| 188 |
+
"learning_rate": 4.19646049338577e-06,
|
| 189 |
+
"loss": 0.4255,
|
| 190 |
+
"step": 1100
|
| 191 |
+
},
|
| 192 |
+
{
|
| 193 |
+
"epoch": 0.39699663415897124,
|
| 194 |
+
"grad_norm": 3.93216609954834,
|
| 195 |
+
"learning_rate": 4.151769753307115e-06,
|
| 196 |
+
"loss": 0.4192,
|
| 197 |
+
"step": 1150
|
| 198 |
+
},
|
| 199 |
+
{
|
| 200 |
+
"epoch": 0.41425735738327435,
|
| 201 |
+
"grad_norm": 3.318143606185913,
|
| 202 |
+
"learning_rate": 4.107079013228459e-06,
|
| 203 |
+
"loss": 0.3984,
|
| 204 |
+
"step": 1200
|
| 205 |
+
},
|
| 206 |
+
{
|
| 207 |
+
"epoch": 0.43151808060757746,
|
| 208 |
+
"grad_norm": 3.0835282802581787,
|
| 209 |
+
"learning_rate": 4.062388273149804e-06,
|
| 210 |
+
"loss": 0.4086,
|
| 211 |
+
"step": 1250
|
| 212 |
+
},
|
| 213 |
+
{
|
| 214 |
+
"epoch": 0.44877880383188057,
|
| 215 |
+
"grad_norm": 2.782550811767578,
|
| 216 |
+
"learning_rate": 4.017697533071148e-06,
|
| 217 |
+
"loss": 0.4114,
|
| 218 |
+
"step": 1300
|
| 219 |
+
},
|
| 220 |
+
{
|
| 221 |
+
"epoch": 0.4660395270561837,
|
| 222 |
+
"grad_norm": 3.2588839530944824,
|
| 223 |
+
"learning_rate": 3.973006792992492e-06,
|
| 224 |
+
"loss": 0.4304,
|
| 225 |
+
"step": 1350
|
| 226 |
+
},
|
| 227 |
+
{
|
| 228 |
+
"epoch": 0.48330025028048673,
|
| 229 |
+
"grad_norm": 3.4788544178009033,
|
| 230 |
+
"learning_rate": 3.928316052913836e-06,
|
| 231 |
+
"loss": 0.4047,
|
| 232 |
+
"step": 1400
|
| 233 |
+
},
|
| 234 |
+
{
|
| 235 |
+
"epoch": 0.5005609735047899,
|
| 236 |
+
"grad_norm": 3.0864784717559814,
|
| 237 |
+
"learning_rate": 3.883625312835181e-06,
|
| 238 |
+
"loss": 0.4185,
|
| 239 |
+
"step": 1450
|
| 240 |
+
},
|
| 241 |
+
{
|
| 242 |
+
"epoch": 0.517821696729093,
|
| 243 |
+
"grad_norm": 2.8312442302703857,
|
| 244 |
+
"learning_rate": 3.838934572756525e-06,
|
| 245 |
+
"loss": 0.4325,
|
| 246 |
+
"step": 1500
|
| 247 |
+
},
|
| 248 |
+
{
|
| 249 |
+
"epoch": 0.517821696729093,
|
| 250 |
+
"eval_gen_len": 54.41937172774869,
|
| 251 |
+
"eval_loss": 0.5411620736122131,
|
| 252 |
+
"eval_rouge1": 50.59,
|
| 253 |
+
"eval_rouge2": 36.52,
|
| 254 |
+
"eval_rougeL": 44.19,
|
| 255 |
+
"eval_rougeLsum": 44.63,
|
| 256 |
+
"eval_runtime": 5226.6165,
|
| 257 |
+
"eval_samples_per_second": 1.096,
|
| 258 |
+
"eval_steps_per_second": 0.274,
|
| 259 |
+
"step": 1500
|
| 260 |
}
|
| 261 |
],
|
| 262 |
"logging_steps": 50,
|
|
|
|
| 271 |
"early_stopping_threshold": 0.0
|
| 272 |
},
|
| 273 |
"attributes": {
|
| 274 |
+
"early_stopping_patience_counter": 1
|
| 275 |
}
|
| 276 |
},
|
| 277 |
"TrainerControl": {
|
|
|
|
| 285 |
"attributes": {}
|
| 286 |
}
|
| 287 |
},
|
| 288 |
+
"total_flos": 5.1502595309568e+16,
|
| 289 |
"train_batch_size": 4,
|
| 290 |
"trial_name": null,
|
| 291 |
"trial_params": null
|