mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-21 11:39:50 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
committed by
AdilZouitine
parent
761a2dbcb3
commit
8e6d5f504c
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user