Disable I experiment prediction.

SINGROUP · Dec 12, 2023 · f3276b8 · f3276b8
1 parent edf0493
commit f3276b8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -11,6 +11,7 @@ dependencies:
   - torchvision
   - torchaudio
   - cuda
+  - ninja
   - matplotlib
   - numpy
   - scipy

diff --git a/papers/ice_structure_discovery/predictions/predict_experiments.py b/papers/ice_structure_discovery/predictions/predict_experiments.py
@@ -65,7 +65,7 @@ def apply_preprocessing(X, real_dim):
         "Ying_Jiang_2_1.npz",
         "Ying_Jiang_2_2.npz",
         "Ying_Jiang_3.npz",
-        "Ying_Jiang_4.npz",
+        # "Ying_Jiang_4.npz", # Takes a lot of video memory
         "Ying_Jiang_5.npz",
         "Ying_Jiang_6.npz",
         "Ying_Jiang_7.npz",
-Original file line number
+Diff line change
@@ Expand Up / @@ -11,6 +11,7 @@ dependencies: @@
       - torchvision
       - torchaudio
       - cuda
+      - ninja
       - matplotlib
       - numpy
       - scipy
@@ Expand Down @@