Skip to content

Why pass 'env' when recording eval video while passing 'obs' when recording train video? #20

@jsrimr

Description

@jsrimr

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions