Update modeling_super_linear.py
Browse files- modeling_super_linear.py +5 -5
modeling_super_linear.py
CHANGED
|
@@ -416,11 +416,11 @@ class superLinear(nn.Module):
|
|
| 416 |
# for param in self.experts[expert_freq].parameters():
|
| 417 |
# param.requires_grad = False
|
| 418 |
|
| 419 |
-
|
| 420 |
-
else:
|
| 421 |
-
|
| 422 |
-
|
| 423 |
-
|
| 424 |
|
| 425 |
|
| 426 |
if configs.misc_moe>0:
|
|
|
|
| 416 |
# for param in self.experts[expert_freq].parameters():
|
| 417 |
# param.requires_grad = False
|
| 418 |
|
| 419 |
+
# self.n_experts = len(self.experts)
|
| 420 |
+
# else:
|
| 421 |
+
# for i in range(self.n_experts):
|
| 422 |
+
# print(f"creating expert {i}")
|
| 423 |
+
# self.experts[str(i)] = self.layer_dict[self.layer_type](self.seq_len, self.pred_len)
|
| 424 |
|
| 425 |
|
| 426 |
if configs.misc_moe>0:
|