#276 Change logging_dir to output_dir (#277)

Nota-NetsPresso · Jul 5, 2024 · 5f622aa · 5f622aa
1 parent ac88e2a
commit 5f622aa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/netspresso/metadata/trainer.py b/netspresso/metadata/trainer.py
@@ -24,7 +24,7 @@ class TrainingInfo:
 @dataclass
 class TrainerMetadata(BaseMetadata):
     task_type: TaskType = TaskType.TRAIN
-    logging_dir: str = ""
+    output_dir: str = ""
     best_fx_model_path: str = ""
     best_onnx_model_path: str = ""
     hparams: str = ""
@@ -48,8 +48,8 @@ def update_training_info(self, epochs, batch_size, learning_rate, optimizer):
     def update_training_result(self, training_summary):
         self.traning_result = training_summary
 
-    def update_logging_dir(self, logging_dir):
-        self.logging_dir = logging_dir
+    def update_output_dir(self, output_dir):
+        self.output_dir = output_dir
 
     def update_best_fx_model_path(self, best_fx_model_path):
         self.best_fx_model_path = best_fx_model_path

diff --git a/netspresso/trainer/trainer.py b/netspresso/trainer/trainer.py
@@ -420,7 +420,7 @@ def train(self, gpus: str, project_name: str, output_dir: Optional[str] = "./out
         destination_folder = Path(output_dir) / project_name
         destination_folder = FileHandler.create_unique_folder(folder_path=destination_folder)
         metadata = TrainerMetadata()
-        metadata.update_logging_dir(logging_dir=Path(destination_folder).resolve().as_posix())
+        metadata.update_output_dir(output_dir=Path(destination_folder).resolve().as_posix())
         metadata.update_model_info(
             task=self.task,
             model=self.model.name,