remove torch.no_grad

mosaicml · Sep 26, 2023 · 63470b4 · 63470b4
1 parent b10beaf
commit 63470b4
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/scripts/eval/eval.py b/scripts/eval/eval.py
@@ -91,7 +91,6 @@ def load_model(model_cfg: DictConfig, tokenizer: PreTrainedTokenizerBase,
                     )
 
 
-@torch.no_grad()
 def evaluate_model(
     model_cfg: DictConfig,
     dist_timeout: Union[float, int],