diff --git a/n3fit/src/n3fit/hyper_optimization/mongofiletrials.py b/n3fit/src/n3fit/hyper_optimization/mongofiletrials.py index a0beb0c21f..be5eb65e02 100644 --- a/n3fit/src/n3fit/hyper_optimization/mongofiletrials.py +++ b/n3fit/src/n3fit/hyper_optimization/mongofiletrials.py @@ -2,6 +2,7 @@ Hyperopt trial object for parallel hyperoptimization with MongoDB. Data are fetched from MongoDB databases and stored in the form of json and tar.gz files within the nnfit folder. """ + import json import logging import os @@ -155,7 +156,7 @@ def __init__( ) self.workers = [] self.log_files = [] - self.output_path = replica_path.parts[-3] + self.job_output_name = replica_path.parts[-3] self._store_trial = False self._json_file = replica_path / "tries.json" @@ -267,7 +268,7 @@ def start_mongo_workers( # my_env["TF_GPU_ALLOCATOR"] = "cuda_malloc_async" # create log files to redirect the mongo-workers output - mongo_workers_logfile = f"mongo-worker_{i+1}_{self.output_path}.log" + mongo_workers_logfile = f"mongo-worker_{i+1}_{self.job_output_name}.log" log_file = open(mongo_workers_logfile, 'w') self.log_files.append(log_file) # run mongo workers