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 3c50756 commit 3746cb4Copy full SHA for 3746cb4
1 file changed
tests/test_memory_stats.py
@@ -11,7 +11,7 @@
11
from MiCoCodeGen import MiCoCodeGen
12
from MiCoUtils import fuse_model
13
14
-def __test_model_memory(model_class, model_kwargs, input_shape, model_name):
+def _test_model_memory(model_class, model_kwargs, input_shape, model_name):
15
"""Test memory optimization for a given model."""
16
print(f"\n{'='*70}")
17
print(f"Testing: {model_name}")
0 commit comments