From 0a96338b13028244474c3b288a8a812aa10fa541 Mon Sep 17 00:00:00 2001 From: Jiang Zhuo <58946806+jiangz17THU@users.noreply.github.com> Date: Fri, 1 Apr 2022 17:55:06 +0800 Subject: [PATCH] [NFC] polish code stype (#626) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: 姜卓 --- colossalai/context/process_group_initializer/initializer_data.py | 1 + 1 file changed, 1 insertion(+) diff --git a/colossalai/context/process_group_initializer/initializer_data.py b/colossalai/context/process_group_initializer/initializer_data.py index aa4e54148..0b8b0d91f 100644 --- a/colossalai/context/process_group_initializer/initializer_data.py +++ b/colossalai/context/process_group_initializer/initializer_data.py @@ -20,6 +20,7 @@ class Initializer_Data(ProcessGroupInitializer): pipeline_parallel_size (int): Size of pipeline parallel. tensor_parallel_size (int): Size of tensor parallel. """ + def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.num_data_parallel_group = self.world_size // self.data_parallel_size