transformers/tests/models/layoutlmv2
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