search

nicklashansen / tdmpc

Code for "Temporal Difference Learning for Model Predictive Control"
MIT License
346 stars 55 forks source link

Why I can't save the video? #7

Closed Bailey-24 closed 2 years ago

Bailey-24 commented 2 years ago

I ran the command python src/train.py task=dog-run save_video=True save_model=True When finished the train, I only got the model.pt, there are not video saved.

My platform is ubuntu20.04

nicklashansen commented 2 years ago

Hi, we only save videos to wandb by default. You'd need to either link a wandb account or modify the line here https://github.com/nicklashansen/tdmpc/blob/96cb7036ecf06f75d5ffd64a0454bbab7d0d3e17/src/logger.py#L76 to use a different library (e.g. imageio) for saving videos. I'm closing this issue but feel free to reopen if you have additional questions.