When I run
python pretrain.py agent=icm domain=walker save_train_video=true,
I encounter an error
File "/root/url_benchmark/video.py", line 83, in record
frame = cv2.resize(obs[-3:].transpose(1, 2, 0),
But recording evaluation video is okay.
Then I found that we call .record method differently between training and evaluating.
When we record eval video,
self.video_recorder.record(self.eval_env)
we pass 'env'.
When we record train video,
self.train_video_recorder.record(time_step.observation)
we pass observation.
Is this intended?