From f106897120dd9f0900a256b61262ca43637a9908 Mon Sep 17 00:00:00 2001 From: Daniel Saavedra Date: Tue, 16 Aug 2022 15:57:00 -0400 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index cf243ba..845b13c 100755 --- a/README.md +++ b/README.md @@ -209,7 +209,7 @@ The configuration file for SSD300 is a json file, which looks like this (exampl `python train_ssd.py -c config.json -o /path/to/result` or -`python train_ssd.py -c config.json -o /path/to/result` +`python train_yolo.py -c config.json -o /path/to/result` By the end of this process, the code will write the weights of the best model to file best_weights.h5 (or whatever name specified in the setting "saved_weights_name" in the config.json file). The training process stops when the loss on the validation set is not improved in 20 consecutive epoches.