diff --git a/Linear_test.py b/Linear_test.py new file mode 100644 index 000000000..0ee1d2083 --- /dev/null +++ b/Linear_test.py @@ -0,0 +1,11 @@ +import paddle,torch,numpy +torch_linear = torch.load("q.pt").cpu() +paddle_linear_state = paddle.load("q.pdparams") +paddle_linear = paddle.nn.Linear(896,896,bias_attr=True) +hidden_states = paddle.load("hidden_states.pdparams") +paddle_linear.set_state_dict(paddle_linear_state) +torch_forward_res = torch_linear(torch.tensor(hidden_states.numpy())) +paddle_forward_res = paddle_linear(hidden_states) +print("torch_forward_res:",torch_forward_res) +print("paddle_forward_res:",paddle_forward_res) +print('allclose_res:',numpy.testing.assert_allclose(torch_forward_res.detach().numpy(),paddle_forward_res)) \ No newline at end of file diff --git a/hidden_states.pdparams b/hidden_states.pdparams new file mode 100644 index 000000000..94c37f8b5 Binary files /dev/null and b/hidden_states.pdparams differ diff --git a/q.pdparams b/q.pdparams new file mode 100644 index 000000000..731d14119 Binary files /dev/null and b/q.pdparams differ diff --git a/q.pt b/q.pt new file mode 100644 index 000000000..4d315c18c Binary files /dev/null and b/q.pt differ