From 3c826e6534aef5ce59f2c1683d0c2676ab400fdd Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Sat, 30 Nov 2024 17:55:09 -0500 Subject: [PATCH] typo --- tests/test_events.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_events.py b/tests/test_events.py index 5a451f6a92..601448e2a2 100644 --- 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='1ep1ba', + max_duration='1ep', optimizers=optimizer, callbacks=[EventCounterCallback()], **kwargs,