EleutherAI / elk

Keeping language models honest by directly eliciting knowledge encoded in their activations.
MIT License
175 stars 32 forks source link

Fix transfer plot bug #280

Closed reaganjlee closed 11 months ago

reaganjlee commented 12 months ago

Previously, elk would go through transfer_dir to find eval.csv files instead of how it responsively finds csv files like in train_dir, which this fixes. This also renames the get_train_dirs to be more clear when using it in both directory situations

derpyplops commented 11 months ago

LGTM!