konfai 1.2.3__tar.gz → 1.2.4__tar.gz

{konfai-1.2.3 → konfai-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: konfai
-Version: 1.2.3
+Version: 1.2.4
 Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
 Author-email: Valentin Boussot <boussot.v@gmail.com>
 License-Expression: Apache-2.0

{konfai-1.2.3 → konfai-1.2.4}/konfai/data/data_manager.py

@@ -317,7 +317,6 @@ class Subset:
                 else:
                     index_set = index_set.intersection(set(self._get_index(s, names)))
                 index = list(index_set)
-                print(index)
         else:
             index = self._get_index(self.subset, names)
         if self.shuffle:
@@ -380,8 +379,8 @@ class Data(ABC):
             use_cache=use_cache,
         )
         self.dataLoader_args = {
-            "num_workers": int(os.environ["KONFAI_WORKERS"]) if use_cache else 0,
-            "pin_memory": True,
+            "num_workers": 0,
+            "pin_memory": False,
         }
         self.data: list[list[dict[str, list[DatasetManager]]]] = []
         self.mapping: list[list[list[tuple[int, int, int]]]] = []

{konfai-1.2.3 → konfai-1.2.4}/konfai/predictor.py

@@ -407,7 +407,6 @@ class _Predictor:
         """
 
         self.model_composite.eval()
-        self.model_composite.to(torch.float32)
         self.model_composite.module.set_state(NetState.PREDICTION)
         self.dataloader_prediction.dataset.load("Prediction")
         with tqdm.tqdm(

{konfai-1.2.3 → konfai-1.2.4}/konfai/trainer.py

@@ -632,7 +632,6 @@ class Trainer(DistributedObject):
             world_size (int): Total number of distributed processes.
         """
         state = State[konfai_state()]
-        print(checkpoints_directory() + self.name + "/")
         if state != State.RESUME and os.path.exists(checkpoints_directory() + self.name + "/"):
             if os.environ["KONFAI_OVERWRITE"] != "True":
                 accept = input(f"The model {self.name} already exists ! Do you want to overwrite it (yes,no) : ")

{konfai-1.2.3 → konfai-1.2.4}/konfai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: konfai
-Version: 1.2.3
+Version: 1.2.4
 Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
 Author-email: Valentin Boussot <boussot.v@gmail.com>
 License-Expression: Apache-2.0

{konfai-1.2.3 → konfai-1.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "konfai"
-version = "1.2.3"
+version = "1.2.4"
 description = "Modular and configurable Deep Learning framework with YAML and PyTorch"
 readme = "README.md"
 requires-python = ">=3.8"