konfai 1.2.0__tar.gz → 1.2.1__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.
- {konfai-1.2.0 → konfai-1.2.1}/PKG-INFO +1 -1
- {konfai-1.2.0 → konfai-1.2.1}/konfai/network/network.py +2 -1
- {konfai-1.2.0 → konfai-1.2.1}/konfai/predictor.py +1 -1
- {konfai-1.2.0 → konfai-1.2.1}/konfai.egg-info/PKG-INFO +1 -1
- {konfai-1.2.0 → konfai-1.2.1}/pyproject.toml +1 -1
- {konfai-1.2.0 → konfai-1.2.1}/LICENSE +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/README.md +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/__init__.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/data/__init__.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/data/augmentation.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/data/data_manager.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/data/patching.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/data/transform.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/evaluator.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/main.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/metric/__init__.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/metric/measure.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/metric/schedulers.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/classification/convNeXt.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/classification/resnet.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/generation/cStyleGan.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/generation/ddpm.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/generation/diffusionGan.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/generation/gan.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/generation/vae.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/registration/registration.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/representation/representation.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/segmentation/NestedUNet.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/models/segmentation/UNet.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/network/__init__.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/network/blocks.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/trainer.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/utils/ITK.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/utils/__init__.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/utils/config.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/utils/dataset.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai/utils/utils.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai.egg-info/SOURCES.txt +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai.egg-info/dependency_links.txt +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai.egg-info/entry_points.txt +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai.egg-info/requires.txt +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/konfai.egg-info/top_level.txt +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/setup.cfg +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/tests/test_config.py +0 -0
- {konfai-1.2.0 → konfai-1.2.1}/tests/test_dataset.py +0 -0
|
@@ -6,7 +6,8 @@ from collections import OrderedDict
|
|
|
6
6
|
from collections.abc import Callable, Iterable, Iterator, Sequence
|
|
7
7
|
from enum import Enum
|
|
8
8
|
from functools import partial
|
|
9
|
-
from typing import Any
|
|
9
|
+
from typing import Any
|
|
10
|
+
from typing_extensions import Self
|
|
10
11
|
|
|
11
12
|
import numpy as np
|
|
12
13
|
import torch
|
|
@@ -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=
|
|
718
|
+
state_dicts.append(torch.load(path + name, weights_only=False))
|
|
719
719
|
else:
|
|
720
720
|
raise Exception(f"Model : {path + name} does not exist !")
|
|
721
721
|
return state_dicts
|
|
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
|
|
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
|