Open yawen-d opened 1 year ago
Assigning to @samuelarnesen to get PR https://github.com/HumanCompatibleAI/imitation/pull/524 over finish line once he starts
Sounds good, feel free to request a review @samuelarnesen once you make some progress. I had already reviewed #524 before.
Problem
scripts.train_rl
,scripts.train_preference_comparisons
,scripts.train_adversarial
andscripts.train_bc
.Solution
record_and_save_video()
function inimitation.util.video_wrapper
that takes in a policy, eval_venv, and a logger to save the video of a policy evaluated on an environment to a designated path.WandbOutputFormat.write()
by adding the following: