From af6920a47680bcf501c5ba808db04b9d561bf75d Mon Sep 17 00:00:00 2001 From: sowmyasri Date: Wed, 28 Aug 2024 12:06:43 +0200 Subject: [PATCH] entrypoint --- examples/Yolo-object-detection/client/entrypoint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/Yolo-object-detection/client/entrypoint.py b/examples/Yolo-object-detection/client/entrypoint.py index 352005ab5..181b5091b 100755 --- a/examples/Yolo-object-detection/client/entrypoint.py +++ b/examples/Yolo-object-detection/client/entrypoint.py @@ -65,7 +65,7 @@ def train(in_model_path, out_model_path, data_path=None, batch_size=32, epochs=1 :param epochs: The number of epochs to train. :type epochs: int """ - os.chdir('darknet') + os.chdir("darknet") darkfile = "example.weights" save_fedn2darknet(in_model_path, darkfile) @@ -83,7 +83,7 @@ def train(in_model_path, out_model_path, data_path=None, batch_size=32, epochs=1 # Darknet executable path darknet_path = "./darknet" # Make sure this path is correct - + print("cwd: ", os.getcwd()) # Command to train YOLO using Darknet command = [darknet_path, "detector", "train", data_file, cfg_file, yolo_converted_weights]