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 0101465 commit 0ae4695Copy full SHA for 0ae4695
tests/unit/autogram/test_engine.py
@@ -124,6 +124,8 @@
124
32,
125
marks=mark.filterwarnings("ignore:There is a performance drop"),
126
),
127
+ (ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly), 32),
128
+ (ModuleFactory(InterModuleParamReuse), 32),
129
(ModuleFactory(FreeParam), 32),
130
(ModuleFactory(NoFreeParam), 32),
131
param(ModuleFactory(Cifar10Model), 16, marks=mark.slow),
@@ -193,8 +195,6 @@ def test_compute_gramian_with_weird_modules(
193
195
"factory",
194
196
[
197
ModuleFactory(ModelUsingSubmoduleParamsDirectly),
- ModuleFactory(ModelAlsoUsingSubmoduleParamsDirectly),
- ModuleFactory(InterModuleParamReuse),
198
],
199
)
200
@mark.parametrize("batch_size", [1, 3, 32])
0 commit comments