Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Migrate CLI to simple_parsing, allowing for YAML based configs #92

Merged
merged 9 commits into from
Feb 20, 2023

Conversation

norabelrose
Copy link
Member

@norabelrose norabelrose commented Feb 20, 2023

This PR migrates the CLI argument parsing and config management to the SimpleParsing library.

Configs are specified using nested dataclasses. The top-level dataclass is RunConfig, which contains all information needed to train a reporter network, including the parameters used to extract hidden states from the model.

Copy link
Collaborator

@AlexTMallen AlexTMallen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Lgtm!

elk/extraction/prompt_dataset.py Outdated Show resolved Hide resolved
@norabelrose norabelrose merged commit 946372d into main Feb 20, 2023
@norabelrose norabelrose deleted the yaml branch February 20, 2023 18:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants