graph LR
Training_Orchestrator["Training Orchestrator"]
Model["Model"]
Loss_Function["Loss Function"]
Optimizer["Optimizer"]
Dataset_DataLoader["Dataset/DataLoader"]
Hooks["Hooks"]
Meters["Meters"]
Distributed_Utilities["Distributed Utilities"]
Training_Orchestrator -- "calls forward pass on" --> Model
Training_Orchestrator -- "computes loss using" --> Loss_Function
Training_Orchestrator -- "triggers optimization on" --> Optimizer
Training_Orchestrator -- "manages data flow from" --> Dataset_DataLoader
Training_Orchestrator -- "invokes" --> Hooks
Training_Orchestrator -- "updates" --> Meters
Training_Orchestrator -- "utilizes" --> Distributed_Utilities
click Training_Orchestrator href "https://github.com/CodeBoarding/GeneratedOnBoardings/blob/main/ClassyVision/Training_Orchestrator.md" "Details"
click Hooks href "https://github.com/CodeBoarding/GeneratedOnBoardings/blob/main/ClassyVision/Hooks.md" "Details"
The Training Orchestrator subsystem in ClassyVision is centered around the ClassificationTask component, which orchestrates the entire machine learning training and evaluation lifecycle.
Training Orchestrator [Expand]
The core orchestrator of the training and evaluation lifecycle for classification tasks. It manages the entire flow, from data loading and model execution to loss computation, optimization, and distributed training coordination. It acts as the central control unit, defining the training and evaluation steps.
Related Classes/Methods:
Represents the neural network model used for predictions.
Related Classes/Methods:
Calculates the discrepancy between model predictions and ground truth.
Related Classes/Methods:
Updates model parameters based on the computed loss.
Related Classes/Methods:
Manages and provides data batches for training and evaluation.
Related Classes/Methods:
Hooks [Expand]
Provides extension points for custom logic at various stages of the training lifecycle.
Related Classes/Methods:
Tracks and aggregates performance metrics during training and evaluation.
Related Classes/Methods:
Provides functionalities for distributed training operations.
Related Classes/Methods: