fix rerun wrong path
This commit is contained in:
parent
605447440f
commit
a1e553f8f1
2
main.py
2
main.py
@ -234,7 +234,7 @@ def main_retrain():
|
|||||||
destination = os.path.join(args.model_destination, "clf.h5")
|
destination = os.path.join(args.model_destination, "clf.h5")
|
||||||
|
|
||||||
logger.info(f"Use command line arguments: {args}")
|
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,
|
domain_tr, flow_tr, name_tr, client_tr, server_windows_tr = dataset.load_or_generate_h5data(args.train_h5data,
|
||||||
args.train_data,
|
args.train_data,
|
||||||
|
Loading…
Reference in New Issue
Block a user