Update README.md

This commit is contained in:
Daniel Saavedra
2022-08-16 15:57:00 -04:00
committed by GitHub
parent 9dab1f1a80
commit f106897120

View File

@@ -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` `python train_ssd.py -c config.json -o /path/to/result`
or 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. 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.