Skip to content

Commit

Permalink
adds early stopping call back (#488)
Browse files Browse the repository at this point in the history
* add early stopping call back

* pre-commit
  • Loading branch information
codestar12 authored Jul 25, 2023
1 parent c6cfb42 commit 8028638
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions llmfoundry/utils/builders.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,9 @@
from typing import Union

from composer import algorithms
from composer.callbacks import (LRMonitor, MemoryMonitor, OptimizerMonitor,
RuntimeEstimator, SpeedMonitor)
from composer.callbacks import (EarlyStopper, LRMonitor, MemoryMonitor,
OptimizerMonitor, RuntimeEstimator,
SpeedMonitor)
from composer.core import Evaluator
from composer.datasets.in_context_learning_evaluation import \
get_icl_task_dataloader
Expand Down Expand Up @@ -57,6 +58,8 @@ def build_callback(name, kwargs):
return MonolithicCheckpointSaver(**kwargs)
elif name == 'scheduled_gc':
return ScheduledGarbageCollector(**kwargs)
elif name == 'early_stopper':
return EarlyStopper(**kwargs)
else:
raise ValueError(f'Not sure how to build callback: {name}')

Expand Down

0 comments on commit 8028638

Please sign in to comment.