razmars commited on
Commit
3eb4fec
·
verified ·
1 Parent(s): 64de4cf

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -492,7 +492,7 @@ class superLinear(nn.Module):
492
  return cycle
493
 
494
 
495
- def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=True):
496
  if len(x_enc.shape) > 2:
497
  x = x_enc.permute(0, 2, 1)
498
  B, V, L = x.shape
 
492
  return cycle
493
 
494
 
495
+ def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
496
  if len(x_enc.shape) > 2:
497
  x = x_enc.permute(0, 2, 1)
498
  B, V, L = x.shape