Skip to content

Commit a4c8e00

Browse files
Kyle1668Quentin-Anthony
authored andcommitted
Removed commented out code
1 parent 6133de9 commit a4c8e00

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

megatron/training.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)