Update

allenai · Oct 22, 2024 · 01b1dc4 · 01b1dc4
1 parent d1c528d
commit 01b1dc4
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/scripts/eval.py b/scripts/eval.py
@@ -263,7 +263,6 @@ def dummy_init_fn(module: torch.nn.Module) -> None:
             log.info("Checkpoint successfully loaded")
             # compute and print the sum of the value of all parameters in the model
             log.info(f"Sum of all parameters: {sum(p.sum().item() for p in dist_model.parameters())}")
-            continue
 
             log.info("Starting evaluating...")
             eval_metrics = trainer.eval()