From b86a654eedc4b7f376243312fbae0994e03985ee Mon Sep 17 00:00:00 2001 From: henrytsui000 Date: Fri, 31 May 2024 16:05:07 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=9D=20[Update]=20Training=20command=20?= =?UTF-8?q?example?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/README.md b/README.md index e63b1c1..456046f 100644 --- a/README.md +++ b/README.md @@ -13,6 +13,28 @@ While the project's structure is still being finalized, we ask that potential co If you are interested in contributing, please keep an eye on project updates or contact us directly at [henrytsui000@gmail.com](mailto:henrytsui000@gmail.com) for more information. +# Installing Dependencies +``` +pip install -r requirements.txt +``` + +# Training the Model +To start training the model, use the following command: +```bash +$python examples/example_train.py +``` +Ensure that you are in the root directory of the project when you run this command. + +The training script will automatically utilize the configuration files from the `yolo/config/` directory as specified by the Hydra configuration framework. This setup allows easy management and adjustment of training parameters without modifying the code. + +To override any configuration parameters directly from the command line, you can specify them after the script. For example, to change the batch size, you can run: + +```bash +$python examples/example_train.py hyper.data.batch_size=8 +``` + + + ## Star History [![Star History Chart](https://api.star-history.com/svg?repos=WongKinYiu/yolov9mit&type=Date)](https://star-history.com/#WongKinYiu/yolov9mit&Date)