From 5ff52f61f08928a984a777f7d293c5f96da66931 Mon Sep 17 00:00:00 2001 From: Vincent Chen Date: Thu, 9 May 2024 13:15:29 -0700 Subject: [PATCH] checker --- composer/utils/misc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/composer/utils/misc.py b/composer/utils/misc.py index ec151c0a3c..73609564ee 100644 --- a/composer/utils/misc.py +++ b/composer/utils/misc.py @@ -65,7 +65,7 @@ def create_interval_scheduler( interval_event = Event.EPOCH_CHECKPOINT if checkpoint_events else Event.EPOCH_END elif time_interval.unit == TimeUnit.ITERATION: interval_event = Event.ITERATION_CHECKPOINT if checkpoint_events else Event.ITERATION_END - elif time_interval.unit in {TimeUnit.BATCH, TimeUnit.TOKEN, TimeUnit.SAMPLE, TimeUnit.DURATION}: + elif time_interval.unit in {TimeUnit.BATCH, TimeUnit.TOKEN, TimeUnit.SAMPLE, TimeUnit.DURATION, TimeUnit.SECOND}: interval_event = Event.BATCH_CHECKPOINT if checkpoint_events else Event.BATCH_END else: raise NotImplementedError(