Disable wandb temporarily, speedup the environment step speed by remove useless a11y tree re-get and terminal output
This commit is contained in:
@@ -42,7 +42,7 @@ def run_single_example(agent, env, example, max_steps, instruction, args, exampl
|
||||
screenshot = __f.read()
|
||||
_f.write(screenshot)
|
||||
# get a11tree and save to wandb
|
||||
thisrun_a11tree = env.controller.get_accessibility_tree()
|
||||
# thisrun_a11tree = env.controller.get_accessibility_tree()
|
||||
# str_table.add_data(wandb.Image(data_or_path=os.path.join(example_result_dir, f"step_{step_idx + 1}_{action_timestamp}.png"), caption=f"step_{step_idx + 1}_{action_timestamp}"),
|
||||
# thisrun_a11tree,
|
||||
# response, action, action_timestamp, done)
|
||||
|
||||
Reference in New Issue
Block a user