ai-training/config.test.yaml at cdcd1f6ea736a26833a3a4a63aff61492e24eb88 - 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 22:26:36 +00:00

Files

T

Oleksandr Bezdieniezhnykh 142c6c4de8 Refactor constants management to use Pydantic BaseModel for configuration

- Replaced module-level path variables in constants.py with a structured Pydantic Config class.
- Updated all relevant modules (train.py, augmentation.py, exports.py, dataset-visualiser.py, manual_run.py) to access paths through the new config structure.
- Fixed bugs related to image processing and model saving.
- Enhanced test infrastructure to accommodate the new configuration approach.

This refactor improves code maintainability and clarity by centralizing configuration management.

2026-03-27 18:18:30 +02:00

12 lines

143 B

YAML

Raw Blame History

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