mirror of
https://github.com/p-e-w/heretic.git
synced 2026-09-27 06:21:29 -07:00
fix: improve print output of scorers
This commit is contained in:
@@ -9,7 +9,6 @@ print_debug_information = true
|
||||
|
||||
batch_size = 2
|
||||
max_response_length = 10
|
||||
kl_divergence_target = 0
|
||||
n_trials = 2
|
||||
n_startup_trials = 1
|
||||
|
||||
@@ -21,11 +20,6 @@ save_directory = "model"
|
||||
|
||||
row_normalization = "none"
|
||||
|
||||
scorers = [
|
||||
{ plugin = "heretic.scorers.keyword_rate.KeywordRate", optimization = "minimize" },
|
||||
{ plugin = "heretic.scorers.kl_divergence.KLDivergence", optimization = "minimize" },
|
||||
]
|
||||
|
||||
[good_prompts]
|
||||
dataset = "mlabonne/harmless_alpaca"
|
||||
commit = "02c6a92cfcf11bb0c387334f8146d149d65b587f"
|
||||
|
||||
Reference in New Issue
Block a user