Skip to content

Commit 8926877

Browse files
committed
Add some tests for FourierFT conv2d layers
1 parent 7b408d8 commit 8926877

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

tests/test_custom_models.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -663,6 +663,27 @@
663663
"init_weights": True,
664664
},
665665
),
666+
(
667+
"Conv2d 1 FourierFT",
668+
"Conv2d",
669+
FourierFTConfig,
670+
{
671+
"target_modules": ["conv2d"],
672+
"n_frequency": 1000,
673+
}
674+
),
675+
(
676+
"Conv2d 2 FourierFT",
677+
"Conv2d",
678+
FourierFTConfig,
679+
{
680+
"target_modules": ["conv2d", "lin0"],
681+
"alpha" : 0.01,
682+
"init_weights": True,
683+
"ifft2_norm": "ortho",
684+
}
685+
),
686+
666687
##########
667688
# VBLoRA #
668689
##########

0 commit comments

Comments
 (0)