for new version

PaddlePaddle · Apr 25, 2021 · 5b95ab5 · 5b95ab5
1 parent 72e2f6d
commit 5b95ab5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/x2paddle/__init__.py b/x2paddle/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "1.0.1"
+__version__ = "1.0.2"
 
 from .core.program import PaddleGraph
 

diff --git a/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py b/x2paddle/op_mapper/dygraph/pytorch2paddle/pytorch_op_mapper.py
@@ -281,7 +281,7 @@ def data(self, graph, node, uid, input_ct):
             scope_name=scope_name,
             data=output_name)
         if self.input_examples is not None:
-            input_np = self.input_examples[input_ct].detach().numpy()
+            input_np = self.input_examples[input_ct].cpu().detach().numpy()
             self.inputs_info[
                 output_name] = [list(input_np.shape), str(input_np.dtype)]
         return [], [output_name]