diff --git a/paddleslim/core/graph_wrapper.py b/paddleslim/core/graph_wrapper.py
index fc0928ef8449d4d40f599296bcdd6f208b840b29..14fc925605df8ed188b94240e38a8b10efa79dd1 100644
--- a/paddleslim/core/graph_wrapper.py
+++ b/paddleslim/core/graph_wrapper.py
@@ -37,7 +37,7 @@ OPTIMIZER_OPS = [
 
 class VarWrapper(object):
     def __init__(self, var, graph):
-        assert isinstance(var, Variable)
+        assert isinstance(var, (Variable, Parameter))
         assert isinstance(graph, GraphWrapper)
         self._var = var
         self._graph = graph