From ad3e63dfaad923f0eba953c4b4bd6753ea34e49b Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Thu, 18 Apr 2024 17:59:27 +0800 Subject: [PATCH 1/2] add dist barrier Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- auto3dseg/algorithm_templates/dints/scripts/infer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/auto3dseg/algorithm_templates/dints/scripts/infer.py b/auto3dseg/algorithm_templates/dints/scripts/infer.py index 6d254903..d57a9ca5 100644 --- a/auto3dseg/algorithm_templates/dints/scripts/infer.py +++ b/auto3dseg/algorithm_templates/dints/scripts/infer.py @@ -138,6 +138,7 @@ def pre_operation(config_file, **override): class InferClass: def __init__(self, config_file: Optional[Union[str, Sequence[str]]] = None, **override): pre_operation(config_file, **override) + dist.barrier() logging.basicConfig(stream=sys.stdout, level=logging.INFO) From fc8a1c649ce66a786613be4d6705b72ad7b3e0be Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Thu, 18 Apr 2024 18:06:38 +0800 Subject: [PATCH 2/2] minor fix Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- auto3dseg/algorithm_templates/dints/scripts/infer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/auto3dseg/algorithm_templates/dints/scripts/infer.py b/auto3dseg/algorithm_templates/dints/scripts/infer.py index d57a9ca5..da075656 100644 --- a/auto3dseg/algorithm_templates/dints/scripts/infer.py +++ b/auto3dseg/algorithm_templates/dints/scripts/infer.py @@ -138,7 +138,8 @@ def pre_operation(config_file, **override): class InferClass: def __init__(self, config_file: Optional[Union[str, Sequence[str]]] = None, **override): pre_operation(config_file, **override) - dist.barrier() + if dist.is_initialized(): + dist.barrier() logging.basicConfig(stream=sys.stdout, level=logging.INFO)