From 23752e7f2ffe387065c3b4069a3f81204066f6c7 Mon Sep 17 00:00:00 2001 From: trias702 <25867060+trias702@users.noreply.github.com> Date: Fri, 12 Jul 2024 18:48:16 -0500 Subject: [PATCH] Fixed minor float check bug in compute_num_steps_per_epoch (#235) Signed-off-by: Daniel Egert --- nemo_aligner/utils/trainer_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo_aligner/utils/trainer_utils.py b/nemo_aligner/utils/trainer_utils.py index b6a17f0d6..524763077 100644 --- a/nemo_aligner/utils/trainer_utils.py +++ b/nemo_aligner/utils/trainer_utils.py @@ -29,7 +29,7 @@ def compute_num_steps_per_epoch( num_steps_per_epoch = sampler.total_samples // sampler.global_batch_size - if limit_train_batches is None or limit_train_batches > 1.0: + if limit_train_batches is None or (isinstance(limit_train_batches, float) and limit_train_batches > 1.0): limit_train_batches = 1.0 if limit_train_batches >= 0: