Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -0
modeling_super_linear.py
CHANGED
|
@@ -292,6 +292,7 @@ class SparseNoisyMoE(nn.Module):
|
|
| 292 |
self.gating_network = nn.Linear(input_dim, self.num_experts, bias=True)
|
| 293 |
|
| 294 |
def get_periodogram(self, inputs, ker_len=50, con=1, n=10000):
|
|
|
|
| 295 |
if inputs.dim() == 2:
|
| 296 |
x_0 = inputs.unsqueeze(2)
|
| 297 |
else:
|
|
|
|
| 292 |
self.gating_network = nn.Linear(input_dim, self.num_experts, bias=True)
|
| 293 |
|
| 294 |
def get_periodogram(self, inputs, ker_len=50, con=1, n=10000):
|
| 295 |
+
n = n / 10
|
| 296 |
if inputs.dim() == 2:
|
| 297 |
x_0 = inputs.unsqueeze(2)
|
| 298 |
else:
|