Skip to content

Commit 039778a

Browse files
committed
🧯fix Wl size test in Fourier Layer
1 parent b8dc3d0 commit 039778a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎test/fourierlayer.jl‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ using Test, Random, Flux
1515
# Wf
1616
@test size(params(FourierLayer(128, 64, 100, 20))[1]) == (128, 64, 20)
1717
# Wl
18-
@test size(params(FourierLayer(128, 64, 100, 20))[2]) == (128, 64, 20)
18+
@test size(params(FourierLayer(128, 64, 100, 20))[2]) == (64, 128)
1919
# bf
2020
@test size(params(FourierLayer(128, 64, 100, 20))[3]) == (1, 64, 20)
2121
# bl

0 commit comments

Comments
 (0)