Fixed a missing (though not important) .eval() line

ExplainableML · Mar 30, 2022 · 92c2032 · 92c2032
1 parent b5f85d6
commit 92c2032
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/language_guidance.py b/language_guidance.py
@@ -232,6 +232,7 @@ def reembed_in_language(language_model, reassigns_topk, device):
  print('Getting language embeddings...')
  unique_labs = {key: None for key in np.unique(reassigns_topk)}
  reembed_collect = []
+ _ = language_model.eval()
  with torch.no_grad():
  language_embeds = language_model(list(unique_labs.keys()), device,
  False).cpu()