konfai 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

konfai/main.py

@@ -2,7 +2,6 @@ import argparse
 import os
 from torch.cuda import device_count
 import torch.multiprocessing as mp
-import submitit
 from konfai.utils.utils import setupAPI, TensorBoard, Log
 
 def main():
@@ -36,7 +35,7 @@ def cluster():
 
         n_gpu = len(config["gpu"].split(","))
         distributedObject.setup(n_gpu*int(config["num_nodes"]))
-
+        import submitit
         executor = submitit.AutoExecutor(folder="./Cluster/")
         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"])
         with TensorBoard(distributedObject.name):

konfai/utils/utils.py

@@ -429,7 +429,6 @@ def setupAPI(parser: argparse.ArgumentParser) -> DistributedObject:
         os.environ["DEEP_LEARNING_API_ROOT"] = "Trainer"
         return Trainer(config=CONFIG_FILE())
 
-import submitit
 
 def setupGPU(world_size: int, port: int, rank: Union[int, None] = None) -> tuple[int , int]:
     try:
@@ -437,6 +436,7 @@ def setupGPU(world_size: int, port: int, rank: Union[int, None] = None) -> tuple
     except:
         host_name = "localhost"
     if rank is None:
+        import submitit
         job_env = submitit.JobEnvironment()
         global_rank = job_env.global_rank
         local_rank = job_env.local_rank

{konfai-1.0.0.dist-info → konfai-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: konfai
-Version: 1.0.0
+Version: 1.0.1
 Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
 Author-email: Valentin Boussot <boussot.v@gmail.com>
 License-Expression: Apache-2.0

{konfai-1.0.0.dist-info → konfai-1.0.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 konfai/__init__.py,sha256=jXMTNml38eX6FSq9d3C_gJVgRLTKHPBUXqOLC7Pqkuo,828
 konfai/evaluator.py,sha256=6YU3bXBy1YcS2kl0YwebkgwXYeO_VoBN59-MLMqj-ds,7468
-konfai/main.py,sha256=Y-8vTgHVecMglB8krGKLbBoFlcoo-Oa4l-PjTDhBzbM,2142
+konfai/main.py,sha256=voh5P5UUY0vjBOsd79O23cVK2LR8WH4c_8ep-kI967E,2149
 konfai/predictor.py,sha256=IOh70fCVm8q-sgZyACNperTO-Vel8QKvYp-FoBY39ao,20236
 konfai/trainer.py,sha256=zGvXd2skcqWgRN9GLx93xYB4Bv-46C0oo7J9My4Levk,16901
 konfai/data/HDF5.py,sha256=QfU8VnyslkQhT_k2AJNFMNkJK7lm75ozxT4WELZt8wk,14390
@@ -30,10 +30,10 @@ konfai/utils/Registration.py,sha256=v1srEBOcgDnHrx0YtsK6bcj0yCMH7wNeaQ3wC7gEvOw,
 konfai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 konfai/utils/config.py,sha256=4NkR1BWXxwtsf95G_xH9_t-pe2unmbRvxzyKCIuS6eE,9894
 konfai/utils/dataset.py,sha256=PNmzxaeFGMBh-pAaR92tDresDjF0QXSli2eGNJyzSVQ,35465
-konfai/utils/utils.py,sha256=BCH7nwDf_Aqt_5lDieBCUEI0HE5TlinG4Z96EOAwA94,20206
-konfai-1.0.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-konfai-1.0.0.dist-info/METADATA,sha256=flcowFH9PD-G7CYRMjBtpX2Dy-XLTY2mCXPfSUL5kAM,1971
-konfai-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-konfai-1.0.0.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
-konfai-1.0.0.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
-konfai-1.0.0.dist-info/RECORD,,
+konfai/utils/utils.py,sha256=hI30DzbpLRRXHyVMW4Kk_2kQo5e-DJ417evGUsCWjXA,20214
+konfai-1.0.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+konfai-1.0.1.dist-info/METADATA,sha256=h1YcCYJ30Bd5oq2LV_78Yz6aJjPKdqd3GDUGyKxGW2M,1971
+konfai-1.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+konfai-1.0.1.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
+konfai-1.0.1.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
+konfai-1.0.1.dist-info/RECORD,,