v4.1 stop log hiddenstate shape

tangminji · Aug 8, 2023 · 97f94b5 · 97f94b5
1 parent d4c9ba6
commit 97f94b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wiki_dataset.py b/wiki_dataset.py
@@ -111,7 +111,7 @@ def __init__(self, classification_head, func, filter=None):
         self.filter = filter
         self.classification_head = classification_head
     def forward(self, hidden_states):
-        print('hidden_state', hidden_states.shape)
+        # print('hidden_state', hidden_states.shape)
         hidden_states = self.func(hidden_states, filter=self.filter)
         hidden_states = self.classification_head(hidden_states)
         return hidden_states