diff --git a/main.py b/main.py index 6c49b35..5332c79 100644 --- a/main.py +++ b/main.py @@ -234,7 +234,7 @@ def main_retrain(): destination = os.path.join(args.model_destination, "clf.h5") logger.info(f"Use command line arguments: {args}") - exists_or_make_path(destination) + exists_or_make_path(args.model_destination) domain_tr, flow_tr, name_tr, client_tr, server_windows_tr = dataset.load_or_generate_h5data(args.train_h5data, args.train_data,