ai-training/config.test.yaml at 18b88ba9bf3e521f9bf7771d8bffe05939201ef3 - ai-training - Gitea: Git with a cup of tea

azaion/ai-training

mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 06:56:34 +00:00

Files

T

Oleksandr Bezdieniezhnykh 18b88ba9bf Refactor configuration and update test structure for improved clarity

- 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.

2026-03-28 07:32:40 +02:00

12 lines

144 B

YAML

Raw Blame History

 training:
   model: 'yolo11n.yaml'
   epochs: 1
   batch: 64
   imgsz: 320
   save_period: 1
   workers: 0
 export:
   onnx_imgsz: 320
   onnx_batch: 1