mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 10:36:35 +00:00
Oleksandr Bezdieniezhnykh
18b88ba9bf
- Updated `.gitignore` to remove committed test fixture data exclusions. - Increased batch size in `config.test.yaml` from 4 to 128 for training. - Simplified directory structure in `config.yaml` by removing unnecessary data paths. - Adjusted paths in `augmentation.py`, `dataset-visualiser.py`, and `exports.py` to align with the new configuration structure. - Enhanced `annotation_queue_handler.py` to utilize the updated configuration for directory management. - Added CSV logging of test results in `conftest.py` for better test reporting. These changes streamline the configuration management and enhance the testing framework, ensuring better organization and clarity in the project.
12 lines
144 B
YAML
12 lines
144 B
YAML
training:
|
|
model: 'yolo11n.yaml'
|
|
epochs: 1
|
|
batch: 64
|
|
imgsz: 320
|
|
save_period: 1
|
|
workers: 0
|
|
|
|
export:
|
|
onnx_imgsz: 320
|
|
onnx_batch: 1
|