Mikkel Denker 70e7d1d656 Fixes torch jit tracing for LayoutLMv2 model (re-open) (#18313) ...

* Fixes torch jit tracing for LayoutLMv2 model.
Pytorch seems to reuse memory for input_shape which caused a mismatch in shapes later in the forward pass.

* Fixed code quality

* avoid unneeded allocation of vector for shape

2022-07-27 06:38:40 -04:00

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_feature_extraction_layoutlmv2.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_layoutlmv2.py

Fixes torch jit tracing for LayoutLMv2 model (re-open) (#18313)

2022-07-27 06:38:40 -04:00

test_processor_layoutlmv2.py

Add LayoutLMv3 (#17060)

2022-05-24 09:53:45 +02:00

test_tokenization_layoutlmv2.py

Fix some typos. (#17560)

2022-07-11 05:00:13 -04:00