konfai 1.2.1__tar.gz → 1.2.2__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.2}/PKG-INFO +1 -1
  2. {konfai-1.2.1 → konfai-1.2.2}/konfai/predictor.py +1 -1
  3. {konfai-1.2.1 → konfai-1.2.2}/konfai.egg-info/PKG-INFO +1 -1
  4. {konfai-1.2.1 → konfai-1.2.2}/pyproject.toml +1 -1
  5. {konfai-1.2.1 → konfai-1.2.2}/LICENSE +0 -0
  6. {konfai-1.2.1 → konfai-1.2.2}/README.md +0 -0
  7. {konfai-1.2.1 → konfai-1.2.2}/konfai/__init__.py +0 -0
  8. {konfai-1.2.1 → konfai-1.2.2}/konfai/data/__init__.py +0 -0
  9. {konfai-1.2.1 → konfai-1.2.2}/konfai/data/augmentation.py +0 -0
  10. {konfai-1.2.1 → konfai-1.2.2}/konfai/data/data_manager.py +0 -0
  11. {konfai-1.2.1 → konfai-1.2.2}/konfai/data/patching.py +0 -0
  12. {konfai-1.2.1 → konfai-1.2.2}/konfai/data/transform.py +0 -0
  13. {konfai-1.2.1 → konfai-1.2.2}/konfai/evaluator.py +0 -0
  14. {konfai-1.2.1 → konfai-1.2.2}/konfai/main.py +0 -0
  15. {konfai-1.2.1 → konfai-1.2.2}/konfai/metric/__init__.py +0 -0
  16. {konfai-1.2.1 → konfai-1.2.2}/konfai/metric/measure.py +0 -0
  17. {konfai-1.2.1 → konfai-1.2.2}/konfai/metric/schedulers.py +0 -0
  18. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/classification/convNeXt.py +0 -0
  19. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/classification/resnet.py +0 -0
  20. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/generation/cStyleGan.py +0 -0
  21. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/generation/ddpm.py +0 -0
  22. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/generation/diffusionGan.py +0 -0
  23. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/generation/gan.py +0 -0
  24. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/generation/vae.py +0 -0
  25. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/registration/registration.py +0 -0
  26. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/representation/representation.py +0 -0
  27. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/segmentation/NestedUNet.py +0 -0
  28. {konfai-1.2.1 → konfai-1.2.2}/konfai/models/segmentation/UNet.py +0 -0
  29. {konfai-1.2.1 → konfai-1.2.2}/konfai/network/__init__.py +0 -0
  30. {konfai-1.2.1 → konfai-1.2.2}/konfai/network/blocks.py +0 -0
  31. {konfai-1.2.1 → konfai-1.2.2}/konfai/network/network.py +0 -0
  32. {konfai-1.2.1 → konfai-1.2.2}/konfai/trainer.py +0 -0
  33. {konfai-1.2.1 → konfai-1.2.2}/konfai/utils/ITK.py +0 -0
  34. {konfai-1.2.1 → konfai-1.2.2}/konfai/utils/__init__.py +0 -0
  35. {konfai-1.2.1 → konfai-1.2.2}/konfai/utils/config.py +0 -0
  36. {konfai-1.2.1 → konfai-1.2.2}/konfai/utils/dataset.py +0 -0
  37. {konfai-1.2.1 → konfai-1.2.2}/konfai/utils/utils.py +0 -0
  38. {konfai-1.2.1 → konfai-1.2.2}/konfai.egg-info/SOURCES.txt +0 -0
  39. {konfai-1.2.1 → konfai-1.2.2}/konfai.egg-info/dependency_links.txt +0 -0
  40. {konfai-1.2.1 → konfai-1.2.2}/konfai.egg-info/entry_points.txt +0 -0
  41. {konfai-1.2.1 → konfai-1.2.2}/konfai.egg-info/requires.txt +0 -0
  42. {konfai-1.2.1 → konfai-1.2.2}/konfai.egg-info/top_level.txt +0 -0
  43. {konfai-1.2.1 → konfai-1.2.2}/setup.cfg +0 -0
  44. {konfai-1.2.1 → konfai-1.2.2}/tests/test_config.py +0 -0
  45. {konfai-1.2.1 → konfai-1.2.2}/tests/test_dataset.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: konfai
3
- Version: 1.2.1
3
+ Version: 1.2.2
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
@@ -715,7 +715,7 @@ class Predictor(DistributedObject):
715
715
  path = models_directory() + self.name + "/StateDict/"
716
716
  name = sorted(os.listdir(path))[-1]
717
717
  if os.path.exists(path + name):
718
- state_dicts.append(torch.load(path + name, weights_only=False))
718
+ state_dicts.append(torch.load(path + name, map_location=torch.device('cpu'), weights_only=False))
719
719
  else:
720
720
  raise Exception(f"Model : {path + name} does not exist !")
721
721
  return state_dicts
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: konfai
3
- Version: 1.2.1
3
+ Version: 1.2.2
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
@@ -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.2"
8
8
  description = "Modular and configurable Deep Learning framework with YAML and PyTorch"
9
9
  readme = "README.md"
10
10
  requires-python = ">=3.8"
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes