parent
8097a56be8
commit
77802f92db
@ -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))
|
||||
Binary file not shown.
Binary file not shown.
Loading…
Reference in new issue