finito
This commit is contained in:
4
main.py
4
main.py
@@ -35,20 +35,20 @@ def main():
|
|||||||
print(f"\n--- Evaluating Initial Architecture ---")
|
print(f"\n--- Evaluating Initial Architecture ---")
|
||||||
red_team_agent.attack_program.eval_program(attacker_prog, trainset)
|
red_team_agent.attack_program.eval_program(attacker_prog, trainset)
|
||||||
|
|
||||||
optimizer = MIPROv2(metric=metric, auto=None)
|
optimizer = MIPROv2(metric=red_team_agent.attack_program.metric, auto=None)
|
||||||
best_prog = optimizer.compile(
|
best_prog = optimizer.compile(
|
||||||
attacker_prog,
|
attacker_prog,
|
||||||
trainset=trainset,
|
trainset=trainset,
|
||||||
max_bootstrapped_demos=2,
|
max_bootstrapped_demos=2,
|
||||||
max_labeled_demos=0,
|
max_labeled_demos=0,
|
||||||
num_trials=3,
|
num_trials=3,
|
||||||
|
num_candidates=6,
|
||||||
)
|
)
|
||||||
|
|
||||||
# evaluating architecture DSPy post-compilation
|
# evaluating architecture DSPy post-compilation
|
||||||
print(f"\n--- Evaluating Optimized Architecture ---")
|
print(f"\n--- Evaluating Optimized Architecture ---")
|
||||||
red_team_agent.attack_program.eval_program(best_prog, trainset)
|
red_team_agent.attack_program.eval_program(best_prog, trainset)
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# push to hub
|
# push to hub
|
||||||
red_team_agent.push_to_hub("farouk1/redteam", commit_message="finito", with_code=True)
|
red_team_agent.push_to_hub("farouk1/redteam", commit_message="finito", with_code=True)
|
||||||
print("---------Pushed to hub!---------")
|
print("---------Pushed to hub!---------")
|
||||||
|
|||||||
Reference in New Issue
Block a user