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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of konfai might be problematic. Click here for more details.

Files changed (7) hide show
  1. konfai/predictor.py +2 -1
  2. {konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/METADATA +1 -1
  3. {konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/RECORD +7 -7
  4. {konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/WHEEL +0 -0
  5. {konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/entry_points.txt +0 -0
  6. {konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/licenses/LICENSE +0 -0
  7. {konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/top_level.txt +0 -0
konfai/predictor.py CHANGED
@@ -407,6 +407,7 @@ class _Predictor:
407
407
  """
408
408
 
409
409
  self.model_composite.eval()
410
+ self.model_composite.to(torch.float32)
410
411
  self.model_composite.module.set_state(NetState.PREDICTION)
411
412
  self.dataloader_prediction.dataset.load("Prediction")
412
413
  with tqdm.tqdm(
@@ -715,7 +716,7 @@ class Predictor(DistributedObject):
715
716
  path = models_directory() + self.name + "/StateDict/"
716
717
  name = sorted(os.listdir(path))[-1]
717
718
  if os.path.exists(path + name):
718
- state_dicts.append(torch.load(path + name, weights_only=False))
719
+ state_dicts.append(torch.load(path + name, map_location=torch.device('cpu'), weights_only=False))
719
720
  else:
720
721
  raise Exception(f"Model : {path + name} does not exist !")
721
722
  return state_dicts
{konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: konfai
3
- Version: 1.2.1
3
+ Version: 1.2.3
4
4
  Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
5
5
  Author-email: Valentin Boussot <boussot.v@gmail.com>
6
6
  License-Expression: Apache-2.0
{konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  konfai/__init__.py,sha256=qjE9Rqxo1sMrkqGS8I5xlGQMZnjIfU-CGgSI5Wmbmbs,1231
2
2
  konfai/evaluator.py,sha256=xAKWUDvdSxqYRUsKqH6ieQF06LWa785aE4zLv4I3_i4,17850
3
3
  konfai/main.py,sha256=Fc4HcJEhPmgunj_f-QYyvQNvjHrKHSUv27Okgu6V5_A,3842
4
- konfai/predictor.py,sha256=-ZcHrFnP7fOUZ4SK4DpNYbir7iScWQnSOZfmeSLtg1I,34598
4
+ konfai/predictor.py,sha256=zFr8d9Hbr3IMdMkVcQLeOj_YtuUhfmf92BWobUfSy8o,34679
5
5
  konfai/trainer.py,sha256=4mc-8r-FxtX_EAn2su8qd-BLLQ0D0So8hh5rmuf6Hqs,27163
6
6
  konfai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
7
  konfai/data/augmentation.py,sha256=vcJE7mosvUkwbpbTN_lGP0S1uJrJYGjlLrt9VnDdJYY,27792
@@ -30,9 +30,9 @@ konfai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
30
30
  konfai/utils/config.py,sha256=a7t44CYMUT5oCDdjL94IswhCVfFbQ5FCgDWZktDDkc4,14347
31
31
  konfai/utils/dataset.py,sha256=Au22fcADKyDJMfS8Z9q8kEXLtKkoufJsH7Pwly6pALo,28288
32
32
  konfai/utils/utils.py,sha256=jCj3tZ8agQYceSY_tlVYp88UFPE5oUn6tXrqnZGrKiI,28410
33
- konfai-1.2.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
34
- konfai-1.2.1.dist-info/METADATA,sha256=UJzi2uPo_JMW4VeEWYm3IB0ZdEaIHbIrdDj4zNNw3WQ,2451
35
- konfai-1.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
- konfai-1.2.1.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
37
- konfai-1.2.1.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
38
- konfai-1.2.1.dist-info/RECORD,,
33
+ konfai-1.2.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
34
+ konfai-1.2.3.dist-info/METADATA,sha256=cwYXLgmo_ZSs1h0d81nP9bwEY6zCewvX33RNlFZUWBE,2451
35
+ konfai-1.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
+ konfai-1.2.3.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
37
+ konfai-1.2.3.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
38
+ konfai-1.2.3.dist-info/RECORD,,
{konfai-1.2.1.dist-info → konfai-1.2.3.dist-info}/WHEEL RENAMED
File without changes