"examples/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "b469e1c1598baac940a2005532344e5992db5ad2"
Unverified Commit dcaa5100 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

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

* [LayoutLMV2 Tests] Make sure input is on GPU

* correct empty line
parent c15bb3fe
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment