Instructions to use hmankar01/pegasus-reddit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use hmankar01/pegasus-reddit with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("hmankar01/pegasus-reddit") model = AutoModelForSeq2SeqLM.from_pretrained("hmankar01/pegasus-reddit", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Training in progress, step 2500, 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 2283652852
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7e7b5dac0c6e968b2eec9d1c2effa7bae254518172157d83df0c94022d0e566d
|
| 3 |
size 2283652852
|
last-checkpoint/optimizer.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 5844445
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7924d43d918924ea0704c375487ea479cec44578e9211c7a91197864aa6bef3d
|
| 3 |
size 5844445
|
last-checkpoint/rng_state.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 14244
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e89ad2fb35d7555ec292dbcb56d3362e45fdc8cdc7a8ae64c80017222dae607f
|
| 3 |
size 14244
|
last-checkpoint/scaler.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 988
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5acc30ad3e6e8cc9d4f01107daa5698ff6a475ad7efa97abbd4c154fc37a4380
|
| 3 |
size 988
|
last-checkpoint/scheduler.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1064
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:22e4a58b00827f494a945fe7f8d3071f9b236011da08d30982cfc8a8a77c1608
|
| 3 |
size 1064
|
last-checkpoint/trainer_state.json
CHANGED
|
@@ -2,9 +2,9 @@
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
-
"epoch":
|
| 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,
|
|
@@ -155,6 +155,41 @@
|
|
| 155 |
"learning_rate": 2.9092827004219407e-05,
|
| 156 |
"loss": 2.9402,
|
| 157 |
"step": 2000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 158 |
}
|
| 159 |
],
|
| 160 |
"logging_steps": 100,
|
|
@@ -174,7 +209,7 @@
|
|
| 174 |
"attributes": {}
|
| 175 |
}
|
| 176 |
},
|
| 177 |
-
"total_flos":
|
| 178 |
"train_batch_size": 4,
|
| 179 |
"trial_name": null,
|
| 180 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 1.0548407508964353,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 2500,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 155 |
"learning_rate": 2.9092827004219407e-05,
|
| 156 |
"loss": 2.9402,
|
| 157 |
"step": 2000
|
| 158 |
+
},
|
| 159 |
+
{
|
| 160 |
+
"epoch": 0.8858890529424173,
|
| 161 |
+
"grad_norm": 3.06837797164917,
|
| 162 |
+
"learning_rate": 2.8037974683544303e-05,
|
| 163 |
+
"loss": 2.9126,
|
| 164 |
+
"step": 2100
|
| 165 |
+
},
|
| 166 |
+
{
|
| 167 |
+
"epoch": 0.9280742459396751,
|
| 168 |
+
"grad_norm": 3.7333426475524902,
|
| 169 |
+
"learning_rate": 2.6993670886075952e-05,
|
| 170 |
+
"loss": 3.0013,
|
| 171 |
+
"step": 2200
|
| 172 |
+
},
|
| 173 |
+
{
|
| 174 |
+
"epoch": 0.9702594389369331,
|
| 175 |
+
"grad_norm": 7.016664028167725,
|
| 176 |
+
"learning_rate": 2.5938818565400847e-05,
|
| 177 |
+
"loss": 2.9823,
|
| 178 |
+
"step": 2300
|
| 179 |
+
},
|
| 180 |
+
{
|
| 181 |
+
"epoch": 1.0126555578991774,
|
| 182 |
+
"grad_norm": 1.065859079360962,
|
| 183 |
+
"learning_rate": 2.4905063291139243e-05,
|
| 184 |
+
"loss": 3.0965,
|
| 185 |
+
"step": 2400
|
| 186 |
+
},
|
| 187 |
+
{
|
| 188 |
+
"epoch": 1.0548407508964353,
|
| 189 |
+
"grad_norm": 0.6410270929336548,
|
| 190 |
+
"learning_rate": 2.3871308016877638e-05,
|
| 191 |
+
"loss": 3.0905,
|
| 192 |
+
"step": 2500
|
| 193 |
}
|
| 194 |
],
|
| 195 |
"logging_steps": 100,
|
|
|
|
| 209 |
"attributes": {}
|
| 210 |
}
|
| 211 |
},
|
| 212 |
+
"total_flos": 8.158690591260672e+16,
|
| 213 |
"train_batch_size": 4,
|
| 214 |
"trial_name": null,
|
| 215 |
"trial_params": null
|