fix args

2026-06-27 16:46:28 +08:00 · 2022-06-06 11:44:19 +02:00
parent 159254348b
commit c99572a6d9
1 changed files with 4 additions and 4 deletions
@@ -170,10 +170,10 @@ class SwitchTransformerEncoderLayer(nn.Module):
        linear = nn.Linear(d_model, dim_feedforward, **factory_kwargs)

        self.linear1 = SwitchFeedForward(
-            capacity_factor,
-            drop_tokens,
-            is_scale_prob,
-            n_experts,
+            capacity_factor=capacity_factor,
+            drop_tokens=drop_tokens,
+            is_scale_prob=is_scale_prob,
+            n_experts=n_experts,
            expert=linear,
            d_model=d_model,
            dim_feedforward=dim_feedforward,