riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[LayoutLMV2 Tests] Make sure input is on GPU (#15314)

Browse Source 
* [LayoutLMV2 Tests] Make sure input is on GPU

* correct empty line
This commit is contained in:
Patrick von Platen 2022-01-24 15:54:47 +01:00 committed by GitHub
parent c15bb3fe19
commit dcaa5100c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_modeling_layoutlmv2.py
View File

@ -122,7 +122,8 @@ class LayoutLMv2ModelTester:
bbox[i, j, 0] = t
image = ImageList(
torch.zeros(self.batch_size, self.num_channels, self.image_size, self.image_size), self.image_size
torch.zeros(self.batch_size, self.num_channels, self.image_size, self.image_size, device=torch_device),
self.image_size,
)
input_mask = None