1 files changed, 1 insertions, 1 deletions

diff --git a/train_lora.py b/train_lora.py
index a9c6e52..6f8644b 100644
--- a/train_lora.py
+++ b/train_lora.py
@@ -630,7 +630,7 @@ def main():
     num_train_epochs = args.num_train_epochs
 
     if num_train_epochs is None:
-        num_train_epochs = math.ceil(len(datamodule.train_dataset) / args.num_train_steps)
+        num_train_epochs = math.ceil(args.num_train_steps / len(datamodule.train_dataset))
 
     optimizer = create_optimizer(
         itertools.chain(