add debug print statement for number of training iterations in Crew class

This commit is contained in:
Vardaan Grover
2025-01-28 14:42:42 +05:30
parent 79daa7061e
commit 91a3d1476d

View File

@@ -491,6 +491,8 @@ class Crew(BaseModel):
"""Trains the crew for a given number of iterations.""" """Trains the crew for a given number of iterations."""
train_crew = self.copy() train_crew = self.copy()
train_crew._setup_for_training(filename) train_crew._setup_for_training(filename)
print("DEBUG N ITERATIONS: ", n_iterations)
for n_iteration in range(n_iterations): for n_iteration in range(n_iterations):
train_crew._train_iteration = n_iteration train_crew._train_iteration = n_iteration