From 552667825b01a89dd212e6bbec4e00c6fed75020 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Wangbo=20Zhao=28=E9=BB=91=E8=89=B2=E6=9E=B7=E9=94=81=29?= <56866854+wangbo-zhao@users.noreply.github.com> Date: Wed, 13 Jul 2022 10:49:06 +0800 Subject: [PATCH] [NFC] polish colossalai/nn/layer/parallel_1d/layers.py code style (#1290) --- colossalai/nn/layer/parallel_1d/layers.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/colossalai/nn/layer/parallel_1d/layers.py b/colossalai/nn/layer/parallel_1d/layers.py index 9459e7139..d2a466e4c 100644 --- a/colossalai/nn/layer/parallel_1d/layers.py +++ b/colossalai/nn/layer/parallel_1d/layers.py @@ -176,7 +176,6 @@ class Classifier1D(ParallelLayer): set_parallel_input(False) env.vocab_parallel = False - def reset_parameters(self, weight_initializer, bias_initializer) -> None: fan_in, fan_out = self.in_features, self.num_classes if self.has_weight: @@ -450,7 +449,6 @@ class Linear1D_Col(ParallelLayer): is_parallel_output = not self.gather_output set_parallel_input(is_parallel_output) - def reset_parameters(self, weight_initializer, bias_initializer) -> None: fan_in, fan_out = self.in_features, self.out_features weight_initializer(self.weight, fan_in=fan_in, fan_out=fan_out) @@ -589,7 +587,6 @@ class Linear1D_Row(ParallelLayer): self._set_tensor_parallel_attributes() set_parallel_input(False) - def reset_parameters(self, weight_initializer, bias_initializer) -> None: fan_in, fan_out = self.in_features, self.out_features weight_initializer(self.weight, fan_in=fan_in, fan_out=fan_out)