Pipeline¶
Pipeline module for managing model training workflow.
pipeline¶
- class openddi.pipeline.Pipeline(args, logger, dataset, model, optimizer)[source]
Bases:
objectPipeline class for managing model training workflow.
Handles model initialization, trainer selection, and feature dimension validation.
- Parameters:
args – Configuration arguments
logger – Logger instance for logging
dataset – Dataset object containing data loaders
model – Neural network model to train
optimizer – Optimizer for training
- __init__(args, logger, dataset, model, optimizer)[source]
- run()[source]
Execute the main training pipeline.
- load_trainer()[source]
Load the appropriate trainer based on model type.
- Returns:
Trainer instance for the specified model
Pipeline¶
- class openddi.pipeline.Pipeline(args, logger, dataset, model, optimizer)[source]¶
Bases:
objectPipeline class for managing model training workflow.
Handles model initialization, trainer selection, and feature dimension validation.
- Parameters:
args – Configuration arguments
logger – Logger instance for logging
dataset – Dataset object containing data loaders
model – Neural network model to train
optimizer – Optimizer for training