diff --git a/roboimi/demos/diana_record_sim_episodes.py b/roboimi/demos/diana_record_sim_episodes.py index b40062e..5eadf79 100644 --- a/roboimi/demos/diana_record_sim_episodes.py +++ b/roboimi/demos/diana_record_sim_episodes.py @@ -21,7 +21,7 @@ def main(): render_cam_name = 'angle' episode_len = 700 #SIM_TASK_CONFIGS[task_name]['episode_len'] - camera_names = ['angle','r_vis'] #SIM_TASK_CONFIGS[task_name]['camera_names'] + camera_names = ['angle','r_vis', 'top'] #SIM_TASK_CONFIGS[task_name]['camera_names'] if task_name == 'sim_transfer': policy = TestPickAndTransferPolicy(inject_noise) print(task_name)