check train.x

issp-center-dev · Oct 26, 2024 · 703536f · 703536f
1 parent 3269e87
commit 703536f
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/abics/applications/latgas_abinitio_interface/aenet_trainer.py b/abics/applications/latgas_abinitio_interface/aenet_trainer.py
@@ -150,17 +150,20 @@ def train(self, train_dir = "train"):
         while True:
             # Repeat until test set error begins to rise
             with open(os.path.join(os.getcwd(), "stdout"), "w") as fi:
-                subprocess.run(
-                    self.train_exe, stdout=fi, stderr=subprocess.STDOUT, check=True
-                )
-                # try:
-                #     subprocess.run(
-                #         self.train_exe, stdout=fi, stderr=subprocess.STDOUT, check=True
-                #     )
-                # except subprocess.CalledProcessError as e:
-                #     print(e.stdout)
-                #     print(e.stderr)
-                #     raise
+                # subprocess.run(
+                #     self.train_exe, stdout=fi, stderr=subprocess.STDOUT, check=True
+                # )
+                try:
+                    # subprocess.run(
+                    #     self.train_exe, stdout=fi, stderr=subprocess.STDOUT, check=True
+                    # )
+                    subprocess.run(
+                        self.train_exe, check=True
+                    )
+                except subprocess.CalledProcessError as e:
+                    print(e.stdout)
+                    print(e.stderr)
+                    raise
             with open("stdout", "r") as trainout:
                 fullout = trainout.readlines()
                 epoch_data = []