Skip to content

Commit

Permalink
entrypoint
Browse files Browse the repository at this point in the history
sowmyasris committed Aug 28, 2024
1 parent d0af95c commit af6920a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/Yolo-object-detection/client/entrypoint.py
Original file line number Diff line number Diff line change
@@ -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]

0 comments on commit af6920a

Please sign in to comment.