Instructions to use harsith05/imdbdata with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use harsith05/imdbdata with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="harsith05/imdbdata")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("harsith05/imdbdata") model = AutoModelForSequenceClassification.from_pretrained("harsith05/imdbdata", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Training in progress, epoch 1, 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 437958648
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:757c0164d062683a7a4c6ff5b9f448c93742b5e19b9f491fa1522e8c7e7a8e1e
|
| 3 |
size 437958648
|
last-checkpoint/optimizer.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:27cd83b39879058c79b84636f93dcd76608027e78b3632e7118bc67c01834baf
|
| 3 |
+
size 4741394
|
last-checkpoint/trainer_state.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"best_global_step": 1563,
|
| 3 |
-
"best_metric": 0.
|
| 4 |
"best_model_checkpoint": "./bert-imdb-cp\\checkpoint-1563",
|
| 5 |
"epoch": 1.0,
|
| 6 |
"eval_steps": 500,
|
|
@@ -11,31 +11,31 @@
|
|
| 11 |
"log_history": [
|
| 12 |
{
|
| 13 |
"epoch": 0.3198976327575176,
|
| 14 |
-
"grad_norm": 1.
|
| 15 |
"learning_rate": 9.361484325015996e-06,
|
| 16 |
-
"loss": 0.
|
| 17 |
"step": 500
|
| 18 |
},
|
| 19 |
{
|
| 20 |
"epoch": 0.6397952655150352,
|
| 21 |
-
"grad_norm":
|
| 22 |
"learning_rate": 8.72168905950096e-06,
|
| 23 |
-
"loss": 0.
|
| 24 |
"step": 1000
|
| 25 |
},
|
| 26 |
{
|
| 27 |
"epoch": 0.9596928982725528,
|
| 28 |
-
"grad_norm":
|
| 29 |
"learning_rate": 8.081893793985925e-06,
|
| 30 |
-
"loss": 0.
|
| 31 |
"step": 1500
|
| 32 |
},
|
| 33 |
{
|
| 34 |
"epoch": 1.0,
|
| 35 |
-
"eval_loss": 0.
|
| 36 |
-
"eval_runtime":
|
| 37 |
-
"eval_samples_per_second":
|
| 38 |
-
"eval_steps_per_second":
|
| 39 |
"step": 1563
|
| 40 |
}
|
| 41 |
],
|
|
|
|
| 1 |
{
|
| 2 |
"best_global_step": 1563,
|
| 3 |
+
"best_metric": 0.2548043727874756,
|
| 4 |
"best_model_checkpoint": "./bert-imdb-cp\\checkpoint-1563",
|
| 5 |
"epoch": 1.0,
|
| 6 |
"eval_steps": 500,
|
|
|
|
| 11 |
"log_history": [
|
| 12 |
{
|
| 13 |
"epoch": 0.3198976327575176,
|
| 14 |
+
"grad_norm": 1.5820907354354858,
|
| 15 |
"learning_rate": 9.361484325015996e-06,
|
| 16 |
+
"loss": 0.2699,
|
| 17 |
"step": 500
|
| 18 |
},
|
| 19 |
{
|
| 20 |
"epoch": 0.6397952655150352,
|
| 21 |
+
"grad_norm": 1.875124454498291,
|
| 22 |
"learning_rate": 8.72168905950096e-06,
|
| 23 |
+
"loss": 0.2901,
|
| 24 |
"step": 1000
|
| 25 |
},
|
| 26 |
{
|
| 27 |
"epoch": 0.9596928982725528,
|
| 28 |
+
"grad_norm": 2.596395492553711,
|
| 29 |
"learning_rate": 8.081893793985925e-06,
|
| 30 |
+
"loss": 0.2749,
|
| 31 |
"step": 1500
|
| 32 |
},
|
| 33 |
{
|
| 34 |
"epoch": 1.0,
|
| 35 |
+
"eval_loss": 0.2548043727874756,
|
| 36 |
+
"eval_runtime": 276.7267,
|
| 37 |
+
"eval_samples_per_second": 90.342,
|
| 38 |
+
"eval_steps_per_second": 5.648,
|
| 39 |
"step": 1563
|
| 40 |
}
|
| 41 |
],
|
last-checkpoint/training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 5432
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bdf14a3601961412e48ff853d34e6894e3d869a6c8607cb63456045f417661ee
|
| 3 |
size 5432
|