fix rerun wrong path

2017-09-29 23:03:07 +02:00 · 2017-09-29 23:03:07 +02:00 · a1e553f8f1
parent 605447440f
commit a1e553f8f1
1 changed files with 1 additions and 1 deletions
--- 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,