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 f7f36be commit 18f86f7Copy full SHA for 18f86f7
framework/e2e/PaddleLT_new/generator/builder_data.py
@@ -75,6 +75,8 @@ def get_single_data(self, framework="paddle"):
75
data.append(torch.tensor(i, requires_grad=True))
76
elif isinstance(i, float):
77
data.append(paddle.to_tensor(i, stop_gradient=False))
78
+ elif isinstance(i, np.float32):
79
+ data.append(paddle.to_tensor(i, stop_gradient=False))
80
elif isinstance(i, int):
81
data.append(paddle.to_tensor(i, stop_gradient=True))
82
else:
0 commit comments