SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
1 changed files with 2 additions and 1 deletions
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 bbox[i, j, 0] = t
image = ImageList( 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 input_mask = None