algorithmic-efficiency icon indicating copy to clipboard operation
algorithmic-efficiency copied to clipboard

Skip eval on train and test for self-reporting results

Open Niccolo-Ajroldi opened this issue 11 months ago • 2 comments

Feature request: allow users to skip eval on train and test

Evaluating on the training and test sets is time-consuming and not necessary for self-reporting results. We should add a flag that allow the user to skip eval on these datasets, to make scoring faster.

Accordingly, in this scenario we should modify:

goals_reached = (
              train_state['validation_goal_reached'] and
              train_state['test_goal_reached'])

into:

goals_reached = (train_state['validation_goal_reached'])

This would speed up self-evalution even more, by stopping training when validation target is reached, avoiding unnecessary usage of computational resources.

Niccolo-Ajroldi avatar Mar 26 '24 12:03 Niccolo-Ajroldi

You're right this is a good suggestion to allow to skip on train and test splits.

priyakasimbeg avatar Mar 29 '24 00:03 priyakasimbeg

We plan to discuss feature requests like these in the benchmark code during the WG meeting on Thursday, 9/5.

fsschneider avatar Sep 04 '24 10:09 fsschneider