fix

135ed1f6 · yuxguo · 84297604 · 135ed1f6
Commit 135ed1f6 authored Mar 10, 2022 by yuxguo
Show whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

model/modules.py
+1 -1

No files found.
--- a/model/modules.py
+++ b/model/modules.py
@@ -413,7 +413,7 @@ class SharedGroupMLP(nn.Module):
        # mlps indicates different experts
        if shared:
            mlps = [MLPModel(group_input_dim, group_output_dim,
-                             hidden_dims=hidden_dims, flatten=False) for i in range(nr_mlps)]
+                             hidden_dims=hidden_dims) for i in range(nr_mlps)]
        else:
            mlps = [GroupMLP(group_input_dim * groups, group_output_dim * groups,
                             hidden_dims=hidden_dims) for i in range(nr_mlps)]