Unverified Commit cf76d94e authored by gregor-soniox's avatar gregor-soniox Committed by GitHub
Browse files

Fix docstring typo (#1118)

parent 5b38de38
......@@ -225,7 +225,7 @@ class ColumnParallelLinear(torch.nn.Module):
in_features: first dimension of matrix A.
out_features: second dimension of matrix A.
bias: If true, add bias
gather_output: If true, call all-gether on output and make Y avaiable
gather_output: If true, call all-gather on output and make Y avaiable
to all GPUs, otherwise, every GPU will have its output
which is Y_i = XA_i
init_method: method to initialize weights. Note that bias is always set
......
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