diff --git a/paddlehub/autofinetune/autoft.py b/paddlehub/autofinetune/autoft.py index ef5d2d6e9492969ce5b2d15787b5f0ea01dad7f4..979c657d9e5573dd497dfa96f8f20f80259cff4e 100644 --- a/paddlehub/autofinetune/autoft.py +++ b/paddlehub/autofinetune/autoft.py @@ -63,7 +63,7 @@ class BaseTuningStrategy(object): self._output_dir = "output_" + time_str else: self._output_dir = output_dir - self.writer = SummaryWriter(logdir=self._output_dir + '/tb_paddle') + self.writer = SummaryWriter(logdir=self._output_dir + '/visualization') @property def thread(self): diff --git a/paddlehub/commands/autofinetune.py b/paddlehub/commands/autofinetune.py index e6401cd97411ce98ef4f28dcef6f76402653fd60..785d56b75c1809499a91fc536ec16d73129f53c4 100644 --- a/paddlehub/commands/autofinetune.py +++ b/paddlehub/commands/autofinetune.py @@ -184,7 +184,7 @@ class AutoFineTuneCommand(BaseCommand): run_round_cnt = run_round_cnt + 1 print("PaddleHub Autofinetune ends.") - with open("./log_file.txt", "w") as f: + with open(autoft._output_dir + "/log_file.txt", "w") as f: best_hparams = evaluator.convert_params(autoft.get_best_hparams()) print("The final best hyperparameters:") f.write("The final best hyperparameters:\n")