We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a8f5380 commit 6bedf56Copy full SHA for 6bedf56
tests/unit/autogram/test_engine.py
@@ -126,6 +126,8 @@
126
32,
127
marks=mark.filterwarnings("ignore:There is a performance drop"),
128
),
129
+ (ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly), 32),
130
+ (ModuleFactory(InterModuleParamReuse), 32),
131
(ModuleFactory(FreeParam), 32),
132
(ModuleFactory(NoFreeParam), 32),
133
param(ModuleFactory(Cifar10Model), 16, marks=mark.slow),
@@ -222,8 +224,6 @@ def test_compute_gramian_with_weird_modules(
222
224
"factory",
223
225
[
226
ModuleFactory(ModelUsingSubmoduleParamsDirectly),
- ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly),
- ModuleFactory(InterModuleParamReuse),
227
],
228
)
229
@mark.parametrize("batch_size", [1, 3, 32])
0 commit comments