Remove num_workers Hotfix (#58)

luxonis · Oct 9, 2024 · 575e0ca · 575e0ca
1 parent 0a7a42c
commit 575e0ca
Showing 1 changed file with 0 additions and 11 deletions.
diff --git a/luxonis_train/utils/config.py b/luxonis_train/utils/config.py
@@ -326,17 +326,6 @@ class Config(LuxonisConfig):
     tuner: TunerConfig | None = None
     ENVIRON: Environ = Field(Environ(), exclude=True)
 
-    @model_validator(mode="after")
-    def validate_num_workers(self) -> Self:
-        if self.loader.name == "LuxonisLoaderTorch":
-            if self.trainer.num_workers != 0:
-                logger.warning(
-                    "Setting `num_workers` to 0 because of "
-                    "compatibility with LuxonisDataset."
-                )
-                self.trainer.num_workers = 0
-        return self
-
     @model_validator(mode="before")
     @classmethod
     def check_environment(cls, data: Any) -> Any: