Update modeling_super_linear.py
Browse files- modeling_super_linear.py +2 -1
modeling_super_linear.py
CHANGED
|
@@ -419,8 +419,9 @@ class Model(nn.Module):
|
|
| 419 |
out, expert_probs = self.moe(x, get_prob=True)
|
| 420 |
else:
|
| 421 |
out = self.moe(x)
|
| 422 |
-
|
| 423 |
if self.train_pred_len < pred_len:
|
|
|
|
| 424 |
outputs = [out]
|
| 425 |
ar_x = torch.cat([x, out], dim=1)[:, -self.seq_len:]
|
| 426 |
for i in range(0, inf_pred_len, self.train_pred_len):
|
|
|
|
| 419 |
out, expert_probs = self.moe(x, get_prob=True)
|
| 420 |
else:
|
| 421 |
out = self.moe(x)
|
| 422 |
+
print("test!")
|
| 423 |
if self.train_pred_len < pred_len:
|
| 424 |
+
print("test2")
|
| 425 |
outputs = [out]
|
| 426 |
ar_x = torch.cat([x, out], dim=1)[:, -self.seq_len:]
|
| 427 |
for i in range(0, inf_pred_len, self.train_pred_len):
|