revert

mosaicml · Nov 30, 2024 · 62f0bea · 62f0bea
1 parent 3c826e6
commit 62f0bea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_events.py b/tests/test_events.py
@@ -57,7 +57,7 @@ def get_trainer(self, precision='fp32', **kwargs):
             precision=precision,
             train_subset_num_batches=self.train_subset_num_batches,
             eval_subset_num_batches=self.eval_subset_num_batches,
-            max_duration='1ep',
+            max_duration='2ep',
             optimizers=optimizer,
             callbacks=[EventCounterCallback()],
             **kwargs,