Work around spconv bug in latest pytorch

Signed-off-by: Kyunggeun Lee <[email protected]>
quic · Feb 16, 2025 · 5359ed5 · 5359ed5
1 parent de5c4b5
commit 5359ed5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TrainingExtensions/torch/src/python/aimet_torch/custom/custom_modules.py b/TrainingExtensions/torch/src/python/aimet_torch/custom/custom_modules.py
@@ -298,4 +298,4 @@ def create_quantizable_sparse_sequential(module: spconv.SparseSequential) -> Qua
         :return: Newly created QuantizableSparseSequential
         """
         # pylint: disable=protected-access
-        return QuantizableSparseSequential(module._modules)
+        return QuantizableSparseSequential(**module._modules)