mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-23 16:54:56 +00:00
add trajectory saving to eval mode.
This commit is contained in:
parent
c421582b6f
commit
7a4edb569c
4 changed files with 144 additions and 4 deletions
|
|
@ -104,6 +104,13 @@ def evaluate_log(
|
|||
writer.write(sample)
|
||||
print(f"Evaluation samples saved to {samples_filepath}")
|
||||
|
||||
try:
|
||||
from atroposlib.frontend.jsonl2html import generate_eval_html
|
||||
|
||||
generate_eval_html(samples_filepath)
|
||||
except Exception as e:
|
||||
print(f"Warning: Failed to generate eval HTML viewer: {e}")
|
||||
|
||||
|
||||
class EvalBase(ABC):
|
||||
""" """
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue