wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

This commit is contained in:
Remi Cadene 2024-03-02 18:11:50 +00:00
parent 661bda45ea
commit 48ded3dbc7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/logger.py
View File

@ -144,5 +144,5 @@ class Logger:
def log_video(self, video, step, mode="train"): def log_video(self, video, step, mode="train"):
assert mode in {"train", "eval"} assert mode in {"train", "eval"}
wandb_video = self._wandb.Video(video, fps=self.cfg.fps, format="mp4") wandb_video = self._wandb.Video(video, fps=self._cfg.fps, format="mp4")
self._wandb.log({f"{mode}/video": wandb_video}, step=step) self._wandb.log({f"{mode}/video": wandb_video}, step=step)