Update modeling_super_linear.py
Browse files- modeling_super_linear.py +2 -0
modeling_super_linear.py
CHANGED
|
@@ -509,6 +509,8 @@ class superLinear(nn.Module):
|
|
| 509 |
else:
|
| 510 |
out, self.moe_loss = self.moe(x)
|
| 511 |
|
|
|
|
|
|
|
| 512 |
|
| 513 |
if self.auto_regressive and self.max_horizon < self.inf_pred_len:
|
| 514 |
#print("bitch")
|
|
|
|
| 509 |
else:
|
| 510 |
out, self.moe_loss = self.moe(x)
|
| 511 |
|
| 512 |
+
print(out.shape)
|
| 513 |
+
|
| 514 |
|
| 515 |
if self.auto_regressive and self.max_horizon < self.inf_pred_len:
|
| 516 |
#print("bitch")
|