konfai 1.2.3__py3-none-any.whl → 1.2.4__py3-none-any.whl

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/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/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.dist-info → konfai-1.2.4.dist-info}/METADATA

@@ -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.dist-info → konfai-1.2.4.dist-info}/RECORD

@@ -1,11 +1,11 @@
 konfai/__init__.py,sha256=qjE9Rqxo1sMrkqGS8I5xlGQMZnjIfU-CGgSI5Wmbmbs,1231
 konfai/evaluator.py,sha256=xAKWUDvdSxqYRUsKqH6ieQF06LWa785aE4zLv4I3_i4,17850
 konfai/main.py,sha256=Fc4HcJEhPmgunj_f-QYyvQNvjHrKHSUv27Okgu6V5_A,3842
-konfai/predictor.py,sha256=zFr8d9Hbr3IMdMkVcQLeOj_YtuUhfmf92BWobUfSy8o,34679
-konfai/trainer.py,sha256=4mc-8r-FxtX_EAn2su8qd-BLLQ0D0So8hh5rmuf6Hqs,27163
+konfai/predictor.py,sha256=S-KlITnyxcAMHmd7aLsoKlJSiZym63yBsJd404HWp9o,34632
+konfai/trainer.py,sha256=g_TkPDUjToFGDGB7aaRZMn-fQllHV_I2GHFKUzDGF8o,27106
 konfai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 konfai/data/augmentation.py,sha256=vcJE7mosvUkwbpbTN_lGP0S1uJrJYGjlLrt9VnDdJYY,27792
-konfai/data/data_manager.py,sha256=7-ruYS2HJMvLzTq5425loOf2OCpAm5MspdYoO74ICsw,31180
+konfai/data/data_manager.py,sha256=I5YG2HIRWQtP_6Lu9KsEWEJzAFFMTMWOYv7dzlcdUN4,31100
 konfai/data/patching.py,sha256=jS35OxnJagKNUnJu7TzuGZpVj9fP-6H4nc2OEYOGgt8,16494
 konfai/data/transform.py,sha256=YCldsqTTBFFCqc_VdvyuNVs2kmV56CxQBN5XhEoPxho,27745
 konfai/metric/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -30,9 +30,9 @@ konfai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 konfai/utils/config.py,sha256=a7t44CYMUT5oCDdjL94IswhCVfFbQ5FCgDWZktDDkc4,14347
 konfai/utils/dataset.py,sha256=Au22fcADKyDJMfS8Z9q8kEXLtKkoufJsH7Pwly6pALo,28288
 konfai/utils/utils.py,sha256=jCj3tZ8agQYceSY_tlVYp88UFPE5oUn6tXrqnZGrKiI,28410
-konfai-1.2.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-konfai-1.2.3.dist-info/METADATA,sha256=cwYXLgmo_ZSs1h0d81nP9bwEY6zCewvX33RNlFZUWBE,2451
-konfai-1.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-konfai-1.2.3.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
-konfai-1.2.3.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
-konfai-1.2.3.dist-info/RECORD,,
+konfai-1.2.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+konfai-1.2.4.dist-info/METADATA,sha256=y505AGDJqf-U3tO8uj9jQ4bo3g_fqdROVA-T4JZ96d8,2451
+konfai-1.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+konfai-1.2.4.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
+konfai-1.2.4.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
+konfai-1.2.4.dist-info/RECORD,,