konfai 1.2.1__tar.gz → 1.2.3__tar.gz

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 (45) hide show
  1. {konfai-1.2.1 → konfai-1.2.3}/PKG-INFO +1 -1
  2. {konfai-1.2.1 → konfai-1.2.3}/konfai/predictor.py +2 -1
  3. {konfai-1.2.1 → konfai-1.2.3}/konfai.egg-info/PKG-INFO +1 -1
  4. {konfai-1.2.1 → konfai-1.2.3}/pyproject.toml +1 -1
  5. {konfai-1.2.1 → konfai-1.2.3}/LICENSE +0 -0
  6. {konfai-1.2.1 → konfai-1.2.3}/README.md +0 -0
  7. {konfai-1.2.1 → konfai-1.2.3}/konfai/__init__.py +0 -0
  8. {konfai-1.2.1 → konfai-1.2.3}/konfai/data/__init__.py +0 -0
  9. {konfai-1.2.1 → konfai-1.2.3}/konfai/data/augmentation.py +0 -0
  10. {konfai-1.2.1 → konfai-1.2.3}/konfai/data/data_manager.py +0 -0
  11. {konfai-1.2.1 → konfai-1.2.3}/konfai/data/patching.py +0 -0
  12. {konfai-1.2.1 → konfai-1.2.3}/konfai/data/transform.py +0 -0
  13. {konfai-1.2.1 → konfai-1.2.3}/konfai/evaluator.py +0 -0
  14. {konfai-1.2.1 → konfai-1.2.3}/konfai/main.py +0 -0
  15. {konfai-1.2.1 → konfai-1.2.3}/konfai/metric/__init__.py +0 -0
  16. {konfai-1.2.1 → konfai-1.2.3}/konfai/metric/measure.py +0 -0
  17. {konfai-1.2.1 → konfai-1.2.3}/konfai/metric/schedulers.py +0 -0
  18. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/classification/convNeXt.py +0 -0
  19. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/classification/resnet.py +0 -0
  20. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/generation/cStyleGan.py +0 -0
  21. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/generation/ddpm.py +0 -0
  22. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/generation/diffusionGan.py +0 -0
  23. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/generation/gan.py +0 -0
  24. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/generation/vae.py +0 -0
  25. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/registration/registration.py +0 -0
  26. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/representation/representation.py +0 -0
  27. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/segmentation/NestedUNet.py +0 -0
  28. {konfai-1.2.1 → konfai-1.2.3}/konfai/models/segmentation/UNet.py +0 -0
  29. {konfai-1.2.1 → konfai-1.2.3}/konfai/network/__init__.py +0 -0
  30. {konfai-1.2.1 → konfai-1.2.3}/konfai/network/blocks.py +0 -0
  31. {konfai-1.2.1 → konfai-1.2.3}/konfai/network/network.py +0 -0
  32. {konfai-1.2.1 → konfai-1.2.3}/konfai/trainer.py +0 -0
  33. {konfai-1.2.1 → konfai-1.2.3}/konfai/utils/ITK.py +0 -0
  34. {konfai-1.2.1 → konfai-1.2.3}/konfai/utils/__init__.py +0 -0
  35. {konfai-1.2.1 → konfai-1.2.3}/konfai/utils/config.py +0 -0
  36. {konfai-1.2.1 → konfai-1.2.3}/konfai/utils/dataset.py +0 -0
  37. {konfai-1.2.1 → konfai-1.2.3}/konfai/utils/utils.py +0 -0
  38. {konfai-1.2.1 → konfai-1.2.3}/konfai.egg-info/SOURCES.txt +0 -0
  39. {konfai-1.2.1 → konfai-1.2.3}/konfai.egg-info/dependency_links.txt +0 -0
  40. {konfai-1.2.1 → konfai-1.2.3}/konfai.egg-info/entry_points.txt +0 -0
  41. {konfai-1.2.1 → konfai-1.2.3}/konfai.egg-info/requires.txt +0 -0
  42. {konfai-1.2.1 → konfai-1.2.3}/konfai.egg-info/top_level.txt +0 -0
  43. {konfai-1.2.1 → konfai-1.2.3}/setup.cfg +0 -0
  44. {konfai-1.2.1 → konfai-1.2.3}/tests/test_config.py +0 -0
  45. {konfai-1.2.1 → konfai-1.2.3}/tests/test_dataset.py +0 -0
{konfai-1.2.1 → konfai-1.2.3}/PKG-INFO 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 → konfai-1.2.3}/konfai/predictor.py RENAMED
@@ -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 → konfai-1.2.3}/konfai.egg-info/PKG-INFO 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 → konfai-1.2.3}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "konfai"
7
- version = "1.2.1"
7
+ version = "1.2.3"
8
8
  description = "Modular and configurable Deep Learning framework with YAML and PyTorch"
9
9
  readme = "README.md"
10
10
  requires-python = ">=3.8"
{konfai-1.2.1 → konfai-1.2.3}/LICENSE RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/README.md RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/__init__.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/data/__init__.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/data/patching.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/data/transform.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/evaluator.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/main.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/metric/__init__.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/metric/measure.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/network/blocks.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/network/network.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/trainer.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/utils/ITK.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/utils/__init__.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/utils/config.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/utils/dataset.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/konfai/utils/utils.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/setup.cfg RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/tests/test_config.py RENAMED
File without changes
{konfai-1.2.1 → konfai-1.2.3}/tests/test_dataset.py RENAMED
File without changes