Update lerobot/scripts/train.py

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
amandip7 2024-06-04 19:46:51 +05:30 committed by GitHub
parent 8ed0181972
commit e4d03066cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -171,7 +171,7 @@ def log_train_info(logger: Logger, info, step, cfg, dataset, is_offline):
f"lr:{lr:0.1e}",
# in seconds
f"updt_s:{update_s:.3f}",
f"dataloading_s:{dataloading_s:.3f}",
f"data_s:{dataloading_s:.3f}", # if not ~0, you are bottlenecked by cpu or io
]
logging.info(" ".join(log_items))