File tree Expand file tree Collapse file tree 1 file changed +1
-6
lines changed Expand file tree Collapse file tree 1 file changed +1
-6
lines changed Original file line number Diff line number Diff line change @@ -1628,11 +1628,6 @@ def train(
16281628 # Evaluation
16291629 is_eval_internal = neox_args .eval_interval and iteration % neox_args .eval_interval == 0
16301630 is_validation_configured = bool (neox_args .do_valid ) or (isinstance (neox_args .eval_tasks , list ) and len (neox_args .eval_tasks ) > 0 )
1631- # if (
1632- # neox_args.eval_interval
1633- # and iteration % neox_args.eval_interval == 0
1634- # # and neox_args.do_valid
1635- # ):
16361631 if is_eval_internal and is_validation_configured :
16371632 prefix = "iteration {}" .format (iteration )
16381633 evaluate_and_print_results (
@@ -1724,7 +1719,7 @@ def evaluate(
17241719 deepspeed .checkpointing .reset ()
17251720
17261721 # reduces losses across processes for logging & run eval harness tasks
1727- eval_results = { "lm_loss" : reduce_losses (losses ).mean ().item ()}
1722+ eval_results [ "lm_loss" ] = reduce_losses (losses ).mean ().item ()
17281723 for key in metric_dicts .keys ():
17291724 eval_results [key ] = reduce_losses (metric_dicts [key ]).mean ().item ()
17301725
You can’t perform that action at this time.
0 commit comments