Skip to content

Commit c7f0da4

Browse files
committed
fix FourierCorrelation
1 parent 4c9bc98 commit c7f0da4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

layers/FourierCorrelation.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from utils.masking import LocalMask
1111
#from layers.mwt import MWT_CZ1d
1212

13+
1314
def get_dynamic_modes(seq_len, modes):
1415
rate1 = seq_len // 96
1516
if rate1 <= 1:
@@ -31,7 +32,7 @@ def get_dynamic_modes(seq_len, modes):
3132

3233
# Cross Fourier Former
3334
class SpectralConvCross1d(nn.Module):
34-
def __init__(self, in_channels, out_channels, seq_len_q, seq_len_kv, modes1=0,policy=0):
35+
def __init__(self, in_channels, out_channels, seq_len_q, seq_len_kv, modes1=0, policy=0):
3536
super(SpectralConvCross1d, self).__init__()
3637
print('corss fourier correlation used!')
3738

0 commit comments

Comments
 (0)