Update modeling_super_linear.py
Browse files- modeling_super_linear.py +3 -0
modeling_super_linear.py
CHANGED
|
@@ -464,6 +464,9 @@ class superLinear(nn.Module):
|
|
| 464 |
else:
|
| 465 |
out, self.moe_loss = self.moe(x)
|
| 466 |
|
|
|
|
|
|
|
|
|
|
| 467 |
if self.auto_regressive and self.max_horizon < self.inf_pred_len:
|
| 468 |
print("bitch")
|
| 469 |
outputs = [out]
|
|
|
|
| 464 |
else:
|
| 465 |
out, self.moe_loss = self.moe(x)
|
| 466 |
|
| 467 |
+
print(F"auto_regressive: {self.auto_regressive}")
|
| 468 |
+
print(F"max_horizon: {self.max_horizon}")
|
| 469 |
+
print(F"self.inf_pred_len: {self.inf_pred_len}")
|
| 470 |
if self.auto_regressive and self.max_horizon < self.inf_pred_len:
|
| 471 |
print("bitch")
|
| 472 |
outputs = [out]
|