henry000 commited on
Commit
b86a654
Β·
1 Parent(s): 9912678

πŸ“ [Update] Training command example

Browse files
Files changed (1) hide show
  1. README.md +22 -0
README.md CHANGED
@@ -13,6 +13,28 @@ While the project's structure is still being finalized, we ask that potential co
13
 
14
  If you are interested in contributing, please keep an eye on project updates or contact us directly at [[email protected]](mailto:[email protected]) for more information.
15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  ## Star History
17
 
18
  [![Star History Chart](https://api.star-history.com/svg?repos=WongKinYiu/yolov9mit&type=Date)](https://star-history.com/#WongKinYiu/yolov9mit&Date)
 
13
 
14
  If you are interested in contributing, please keep an eye on project updates or contact us directly at [[email protected]](mailto:[email protected]) for more information.
15
 
16
+ # Installing Dependencies
17
+ ```
18
+ pip install -r requirements.txt
19
+ ```
20
+
21
+ # Training the Model
22
+ To start training the model, use the following command:
23
+ ```bash
24
+ $python examples/example_train.py
25
+ ```
26
+ Ensure that you are in the root directory of the project when you run this command.
27
+
28
+ 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.
29
+
30
+ 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:
31
+
32
+ ```bash
33
+ $python examples/example_train.py hyper.data.batch_size=8
34
+ ```
35
+
36
+
37
+
38
  ## Star History
39
 
40
  [![Star History Chart](https://api.star-history.com/svg?repos=WongKinYiu/yolov9mit&type=Date)](https://star-history.com/#WongKinYiu/yolov9mit&Date)