From 92c20327310d94182ef4ab9499960f18005fcc8a Mon Sep 17 00:00:00 2001
From: Karsten Roth <karsten.rh1@gmail.com>
Date: Wed, 30 Mar 2022 14:59:05 +0200
Subject: [PATCH] Fixed a missing (though not important) .eval() line

---
 language_guidance.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/language_guidance.py b/language_guidance.py
index 2fe71c7..66b887c 100644
--- 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()