[fix] rm comments;

pull/6069/head
duanjunwen 2 months ago
parent a92e16719b
commit 45f17fc6cc

@ -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,

Loading…
Cancel
Save