HUSTAI / uie_pytorch

PaddleNLP UIE模型的PyTorch版实现
Apache License 2.0
539 stars 93 forks source link

uie-base 转torch,验证时报错 #43

Open DroLuo opened 2 months ago

DroLuo commented 2 months ago

`[2024-04-16 12:03:00,092] [ INFO] - Validating PyTorch model...

W0416 12:03:00.840634 21336 gpu_resources.cc:275] WARNING: device: . The installed Paddle is compiled with CUDNN 8.6, but CUDNN version in your machine is 8.1, which may cause serious incompatible bug. Please recompile or reinstall Paddle with compatible CUDNN version.

[2024-04-16 12:03:03,910] [ INFO] - -[✓] Pytorch model output names match reference model ({'start_prob', 'end_prob'})

[2024-04-16 12:03:03,930] [ INFO] - - Validating PyTorch Model output "start_prob":

[2024-04-16 12:03:03,934] [ INFO] - -[✓] (1, 512) matches (1, 512)

[2024-04-16 12:03:03,943] [ INFO] - -[x] values not close enough (atol: 1e-05)

Traceback (most recent call last): File ".\convert.py", line 478, in do_main() File ".\convert.py", line 462, in do_main validate_model(tokenizer, model, paddle_model, model_type) File ".\convert.py", line 424, in validate_model "Outputs values doesn't match between reference model and Pytorch converted model: " ValueError: Outputs values doesn't match between reference model and Pytorch converted model: Got max absolute difference of: 0.6232820749282837`