add docs and minor fix

README.md

@@ -567,6 +567,7 @@ model.tune()
 - [**Callbacks**](https://github.com/luxonis/luxonis-train/blob/main/luxonis_train/callbacks/README.md): Allow custom code to be executed at different stages of training.
 - [**Optimizers**](https://github.com/luxonis/luxonis-train/blob/main/configs/README.md#optimizer): Control how the model's weights are updated.
 - [**Schedulers**](https://github.com/luxonis/luxonis-train/blob/main/configs/README.md#scheduler): Adjust the learning rate during training.
+- [**Training Strategy**](https://github.com/luxonis/luxonis-train/blob/main/configs/README.md#training-strategy): Specify a custom combination of optimizer and scheduler to tailor the training process for specific use cases.
 
 **Creating Custom Components:**
 
@@ -581,6 +582,7 @@ Registered components can be referenced in the config file. Custom components ne
 - **Callbacks** - [`lightning.pytorch.callbacks.Callback`](https://lightning.ai/docs/pytorch/stable/extensions/callbacks.html), requires manual registration to the `CALLBACKS` registry
 - **Optimizers** - [`torch.optim.Optimizer`](https://pytorch.org/docs/stable/optim.html#torch.optim.Optimizer), requires manual registration to the `OPTIMIZERS` registry
 - **Schedulers** - [`torch.optim.lr_scheduler.LRScheduler`](https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate), requires manual registration to the `SCHEDULERS` registry
+- **Training Strategy** - [`BaseTrainingStrategy`](https://github.com/luxonis/luxonis-train/blob/main/luxonis_train/strategies/base_strategy.py)
 
 **Examples:**
 

luxonis_train/config/config.py

@@ -337,7 +337,7 @@ class SchedulerConfig(BaseModelExtraForbid):
 
 
 class TrainingStrategyConfig(BaseModelExtraForbid):
-    name: str = "TripleLRSGDStrategy"
+    name: str
     params: Params = {}
 
 
@@ -387,7 +387,7 @@ class TrainerConfig(BaseModelExtraForbid):
 
     optimizer: OptimizerConfig = OptimizerConfig()
     scheduler: SchedulerConfig = SchedulerConfig()
-    training_strategy: TrainingStrategyConfig = TrainingStrategyConfig()
+    training_strategy: TrainingStrategyConfig | None = None
 
     @model_validator(mode="after")
     def validate_deterministic(self) -> Self:

luxonis_train/models/luxonis_lightning.py

@@ -273,7 +273,15 @@ def __init__(
 
         self.load_checkpoint(self.cfg.model.weights)
 
-        if self.cfg.trainer.training_strategy.params:
+        if self.cfg.trainer.training_strategy is not None:
+            if self.cfg.trainer.optimizer is not None:
+                logger.warning(
+                    "Training strategy is active; the specified optimizer will be ignored."
+                )
+            if self.cfg.trainer.scheduler is not None:
+                logger.warning(
+                    "Training strategy is active; the specified scheduler will be ignored."
+                )
             self.training_strategy = STRATEGIES.get(
                 self.cfg.trainer.training_strategy.name
             )(