diff --git a/auto3dseg/algorithm_templates/dints/scripts/infer.py b/auto3dseg/algorithm_templates/dints/scripts/infer.py index 6d254903..da075656 100644 --- a/auto3dseg/algorithm_templates/dints/scripts/infer.py +++ b/auto3dseg/algorithm_templates/dints/scripts/infer.py @@ -138,6 +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) + if dist.is_initialized(): + dist.barrier() logging.basicConfig(stream=sys.stdout, level=logging.INFO)