Skip to content

Commit f8a816d

Browse files
committed
update tutorial reqs
1 parent c3eb113 commit f8a816d

File tree

3 files changed

+8
-6
lines changed

3 files changed

+8
-6
lines changed

docs/requirements.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,9 @@ docutils>=0.16
66
sphinx-paramlinks>=0.4.0
77
ipython[notebook]
88

9+
# temporarily included until hub available to evaluate finetuning_scheduler
10+
git+git://github.com/speediedan/pytorch-lightning.git@24d3e43568814ec381ac5be91627629808d62081#egg=pytorch-lightning
11+
912
https://github.com/PyTorchLightning/lightning_sphinx_theme/archive/master.zip#egg=pt-lightning-sphinx-theme
1013

1114
-r ../.actions/requirements.txt

lightning_examples/finetuning-scheduler/.meta.yml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,12 @@ tags:
99
description: |
1010
This notebook introduces the FinetuningScheduler callback and demonstrates the use of FinetuningScheduler to finetune
1111
a small foundational model on the [RTE](https://huggingface.co/datasets/viewer/?dataset=super_glue&config=rte) task
12-
of [SuperGLUE](https://super.gluebenchmark.com/) with iterative earlystopping defined according to a user-specified
13-
schedule. It uses HuggingFace's ``datasets`` and ``transformers`` libraries to retrieve the relevant benchmark data and
14-
foundational model weights.
12+
of [SuperGLUE](https://super.gluebenchmark.com/) with iterative early-stopping defined according to a user-specified
13+
schedule. It uses HuggingFace's ``datasets`` and ``transformers`` libraries to retrieve the relevant benchmark data
14+
and foundational model weights.
1515
requirements:
1616
- transformers
1717
- datasets
18+
- scikit-learn
1819
accelerator:
1920
- GPU

lightning_examples/finetuning-scheduler/finetuning-scheduler.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@
8484
# ```
8585

8686
# %% [markdown]
87-
# ## EarlyStopping and Epoch-Driven Phase Transition Criteria
87+
# ## Early-Stopping and Epoch-Driven Phase Transition Criteria
8888
#
8989
#
9090
# By default, ``FTSEarlyStopping`` and epoch-driven
@@ -396,7 +396,6 @@ def __init__(
396396
model_cfg: Optional[Dict[str, Any]] = None,
397397
task_name: str = DEFAULT_TASK,
398398
experiment_tag: str = "default",
399-
plot_liveloss: bool = False,
400399
):
401400
"""
402401
Args:
@@ -414,7 +413,6 @@ def __init__(
414413
super().__init__()
415414
self.optimizer_init = optimizer_init
416415
self.lr_scheduler_init = lr_scheduler_init
417-
self.plot_liveloss = plot_liveloss
418416
self.pl_lrs_cfg = pl_lrs_cfg or {}
419417
if task_name in TASK_NUM_LABELS.keys():
420418
self.task_name = task_name

0 commit comments

Comments
 (0)