Skip to content

Henryh/layer exit evals#515

Open
Hgherzog wants to merge 4 commits intomainfrom
henryh/layer-exit-evals
Open

Henryh/layer exit evals#515
Hgherzog wants to merge 4 commits intomainfrom
henryh/layer-exit-evals

Conversation

@Hgherzog
Copy link
Copy Markdown
Collaborator

Summary

Add support for evaluating intermediate encoder layer features in downstream evals and fold layer sweeps into the normal full_eval_sweep.py entrypoint.

This adds a feature_exit_depth option to downstream eval tasks, uses token_exit_cfg in the eval wrapper to extract features from a specified encoder depth, and adds --task_names / --layer_depths flags to full_eval_sweep.py so teammates can run layer sweeps on arbitrary checkpoints.

layer_eval_val_curves

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant