Skip to content

Commit

Permalink
Fix output_dir parameter in metadata initialization
Browse files Browse the repository at this point in the history
  • Loading branch information
Only-bottle committed Aug 6, 2024
1 parent dd11ca1 commit 14daa3b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion netspresso/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -527,7 +527,7 @@ def train(self, gpus: str, project_name: str, output_dir: Optional[str] = "./out
project_name = project_name if project_name else f"{self.task}_{self.model_name}".lower()
destination_folder = Path(output_dir) / project_name
destination_folder = FileHandler.create_unique_folder(folder_path=destination_folder)
metadata = self.initialize_metadata(output_dir=output_dir)
metadata = self.initialize_metadata(output_dir=destination_folder)

try:
self.logging.output_dir = output_dir
Expand Down

0 comments on commit 14daa3b

Please sign in to comment.