mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 09:26:36 +00:00
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.
This commit is contained in:
@@ -0,0 +1,11 @@
|
||||
training:
|
||||
model: 'yolo11n.yaml'
|
||||
epochs: 1
|
||||
batch: 4
|
||||
imgsz: 320
|
||||
save_period: 1
|
||||
workers: 0
|
||||
|
||||
export:
|
||||
onnx_imgsz: 320
|
||||
onnx_batch: 1
|
||||
Reference in New Issue
Block a user