Skip to content

Commit

Permalink
πŸ“ [Update] Training command example
Browse files Browse the repository at this point in the history
  • Loading branch information
henrytsui000 committed May 31, 2024
1 parent 9912678 commit b86a654
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit b86a654

Please sign in to comment.