AIMANE 1.1.0 - Quality of Life & Add Finish accuracy settings (Hotfix３)

* [Server] Fix bug when pressing Train before Prepare dataset (Blank dataset) will give error
Nicezki · Jul 31, 2023 · ae66dbb · ae66dbb
1 parent 7e7bacd
commit ae66dbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/aimane.py b/app/aimane.py
@@ -621,7 +621,7 @@ def load_mnist(self, path, usercontent=False):
     def train(self):
         self.sysmane.write_status("Training model starting...", stage="Training model", percentage=0)
         # Check if the dataset is prepared and have dataset in it. check if count.txt exists inside the dataset pa
-        if not os.path.exists(self.dataset_path or not os.path.exists("{}/dataset".format(self.dataset_path or not os.path.exists("{}/count.txt".format(self.dataset_path))))):
+        if not os.path.exists(self.dataset_path) or not os.path.exists("{}/dataset".format(self.dataset_path)) or not os.path.exists("{}/count.txt".format(self.dataset_path)):
             self.sysmane.write_status("[ERROR] Dataset is not prepared.")
             return "Dataset is not prepared."
         else: