Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "1d2c3f6b384cf39e9436b5dfdc32c471fd7fc551"
Commit 55266782 authored by Wangbo Zhao(黑色枷锁)'s avatar Wangbo Zhao(黑色枷锁) Committed by Frank Lee
Browse files

[NFC] polish colossalai/nn/layer/parallel_1d/layers.py code style (#1290)

parent d6f5ef88
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
colossalai/nn/layer/parallel_1d/layers.py
View file @ 55266782
...@@ -176,7 +176,6 @@ class Classifier1D(ParallelLayer): ...@@ -176,7 +176,6 @@ class Classifier1D(ParallelLayer):
set_parallel_input(False) set_parallel_input(False)
env.vocab_parallel = False env.vocab_parallel = False
def reset_parameters(self, weight_initializer, bias_initializer) -> None: def reset_parameters(self, weight_initializer, bias_initializer) -> None:
fan_in, fan_out = self.in_features, self.num_classes fan_in, fan_out = self.in_features, self.num_classes
if self.has_weight: if self.has_weight:
...@@ -450,7 +449,6 @@ class Linear1D_Col(ParallelLayer): ...@@ -450,7 +449,6 @@ class Linear1D_Col(ParallelLayer):
is_parallel_output = not self.gather_output is_parallel_output = not self.gather_output
set_parallel_input(is_parallel_output) set_parallel_input(is_parallel_output)
def reset_parameters(self, weight_initializer, bias_initializer) -> None: def reset_parameters(self, weight_initializer, bias_initializer) -> None:
fan_in, fan_out = self.in_features, self.out_features fan_in, fan_out = self.in_features, self.out_features
weight_initializer(self.weight, fan_in=fan_in, fan_out=fan_out) weight_initializer(self.weight, fan_in=fan_in, fan_out=fan_out)
...@@ -589,7 +587,6 @@ class Linear1D_Row(ParallelLayer): ...@@ -589,7 +587,6 @@ class Linear1D_Row(ParallelLayer):
self._set_tensor_parallel_attributes() self._set_tensor_parallel_attributes()
set_parallel_input(False) set_parallel_input(False)
def reset_parameters(self, weight_initializer, bias_initializer) -> None: def reset_parameters(self, weight_initializer, bias_initializer) -> None:
fan_in, fan_out = self.in_features, self.out_features fan_in, fan_out = self.in_features, self.out_features
weight_initializer(self.weight, fan_in=fan_in, fan_out=fan_out) weight_initializer(self.weight, fan_in=fan_in, fan_out=fan_out)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment