revert model path

Atashnezhad · Jul 4, 2023 · 49c5d85 · 49c5d85
1 parent b6147a5
commit 49c5d85
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/neural_network_model/transfer_learning.py b/neural_network_model/transfer_learning.py
@@ -475,9 +475,6 @@ def train_model(self, epochs=10, batch_size=32, **kwargs):
             model_save_location: location to save the model default is self.model_save_location
         """
         if kwargs.get("model_save_location"):
-            # check if the path exists if not create it
-            if not os.path.exists(kwargs.get("model_save_location")):
-                os.makedirs(kwargs.get("model_save_location"))
             self.model_save_path = kwargs.get("model_save_path")
         model_name = kwargs.get("model_name", "tf_model.h5")