tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix

Browse Source
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
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)