Update keras_test.py

EnzymeAD · Nov 11, 2024 · 6416e6e · 6416e6e
1 parent 2154129
commit 6416e6e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/keras_test.py b/test/keras_test.py
@@ -77,8 +77,8 @@ def main(argv):
         benchfns += [
             ("bert_fit", benchmark.bert.bert_fit_run, Both),
         ]
-
-    if True:
+    # also oom
+    if False:
         benchfns += [
             ("sam_predict", benchmark.sam.sam_predict_run, Both),
             ("sam_fit", benchmark.sam.sam_fit_run, Both),