Update lerobot/scripts/train.py
Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
parent
8ed0181972
commit
e4d03066cf
|
@ -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))
|
||||
|
||||
|
|
Loading…
Reference in New Issue