[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-03-24 13:16:38 +00:00
committed by AdilZouitine
parent 761a2dbcb3
commit 8e6d5f504c
97 changed files with 1596 additions and 492 deletions
+2 -2
View File
@@ -426,7 +426,7 @@ def train(
# Training loop with validation and checkpointing
for epoch in range(cfg.training.num_epochs):
logging.info(f"\nEpoch {epoch+1}/{cfg.training.num_epochs}")
logging.info(f"\nEpoch {epoch + 1}/{cfg.training.num_epochs}")
train_epoch(
model,
@@ -470,7 +470,7 @@ def train(
policy=model,
optimizer=optimizer,
scheduler=None,
identifier=f"{epoch+1:06d}",
identifier=f"{epoch + 1:06d}",
)
step += len(train_loader)