Yih-Dar 4cd7ed4b3b Fix a TF Vision Encoder Decoder test (#15896) ...

* send PyTorch inputs to the correct device

* Fix: TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

2022-03-03 13:21:31 +01:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_modeling_flax_vision_encoder_decoder.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_modeling_tf_vision_encoder_decoder.py

Fix a TF Vision Encoder Decoder test (#15896)

2022-03-03 13:21:31 +01:00

test_modeling_vision_encoder_decoder.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00