konfai 1.0.0__tar.gz → 1.0.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.0.0 → konfai-1.0.1}/PKG-INFO +1 -1
- {konfai-1.0.0 → konfai-1.0.1}/konfai/main.py +1 -2
- {konfai-1.0.0 → konfai-1.0.1}/konfai/utils/utils.py +1 -1
- {konfai-1.0.0 → konfai-1.0.1}/konfai.egg-info/PKG-INFO +1 -1
- {konfai-1.0.0 → konfai-1.0.1}/pyproject.toml +1 -1
- {konfai-1.0.0 → konfai-1.0.1}/LICENSE +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/README.md +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/__init__.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/data/HDF5.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/data/__init__.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/data/augmentation.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/data/dataset.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/data/transform.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/evaluator.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/metric/__init__.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/metric/measure.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/metric/schedulers.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/classification/convNeXt.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/classification/resnet.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/generation/cStyleGan.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/generation/ddpm.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/generation/diffusionGan.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/generation/gan.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/generation/vae.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/registration/registration.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/representation/representation.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/segmentation/NestedUNet.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/models/segmentation/UNet.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/network/__init__.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/network/blocks.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/network/network.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/predictor.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/trainer.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/utils/ITK.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/utils/Registration.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/utils/__init__.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/utils/config.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai/utils/dataset.py +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai.egg-info/SOURCES.txt +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai.egg-info/dependency_links.txt +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai.egg-info/entry_points.txt +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai.egg-info/requires.txt +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/konfai.egg-info/top_level.txt +0 -0
- {konfai-1.0.0 → konfai-1.0.1}/setup.cfg +0 -0
|
@@ -2,7 +2,6 @@ import argparse
|
|
|
2
2
|
import os
|
|
3
3
|
from torch.cuda import device_count
|
|
4
4
|
import torch.multiprocessing as mp
|
|
5
|
-
import submitit
|
|
6
5
|
from konfai.utils.utils import setupAPI, TensorBoard, Log
|
|
7
6
|
|
|
8
7
|
def main():
|
|
@@ -36,7 +35,7 @@ def cluster():
|
|
|
36
35
|
|
|
37
36
|
n_gpu = len(config["gpu"].split(","))
|
|
38
37
|
distributedObject.setup(n_gpu*int(config["num_nodes"]))
|
|
39
|
-
|
|
38
|
+
import submitit
|
|
40
39
|
executor = submitit.AutoExecutor(folder="./Cluster/")
|
|
41
40
|
executor.update_parameters(name=config["name"], mem_gb=config["memory"], gpus_per_node=n_gpu, tasks_per_node=n_gpu//distributedObject.size, cpus_per_task=config["num_workers"], nodes=config["num_nodes"], timeout_min=config["time_limit"])
|
|
42
41
|
with TensorBoard(distributedObject.name):
|
|
@@ -429,7 +429,6 @@ def setupAPI(parser: argparse.ArgumentParser) -> DistributedObject:
|
|
|
429
429
|
os.environ["DEEP_LEARNING_API_ROOT"] = "Trainer"
|
|
430
430
|
return Trainer(config=CONFIG_FILE())
|
|
431
431
|
|
|
432
|
-
import submitit
|
|
433
432
|
|
|
434
433
|
def setupGPU(world_size: int, port: int, rank: Union[int, None] = None) -> tuple[int , int]:
|
|
435
434
|
try:
|
|
@@ -437,6 +436,7 @@ def setupGPU(world_size: int, port: int, rank: Union[int, None] = None) -> tuple
|
|
|
437
436
|
except:
|
|
438
437
|
host_name = "localhost"
|
|
439
438
|
if rank is None:
|
|
439
|
+
import submitit
|
|
440
440
|
job_env = submitit.JobEnvironment()
|
|
441
441
|
global_rank = job_env.global_rank
|
|
442
442
|
local_rank = job_env.local_rank
|
|
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
|