github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-10 20:32:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fix] rm comments;

Browse Source
This commit is contained in:
duanjunwen 2024-09-26 06:13:56 +00:00
parent a92e16719b
commit 45f17fc6cc
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_pipeline/test_schedule/test_zerobubble_pp.py
View File

@ -31,9 +31,6 @@ class MlpModel(nn.Module):
):
super().__init__()
self.layers = nn.Sequential(*[nn.Linear(in_dim, out_dim, bias=None) for _ in range(num_layers)])
# self.layers = nn.ModuleList([nn.Linear(in_dim, out_dim, bias=None) for _ in range(num_layers)])
# if stage_mgr:
# self.held_layers = self.layers[stage_index[0]:stage_index[1]]
def forward(
self,