transformers
74297d0a - [Switch Transformers] Fix failing slow test (#20346)

Commit

3 years ago

[Switch Transformers] Fix failing slow test (#20346) * run slow test on GPU * remove unnecessary device assignment * use `torch_device` instead

References

Author

younesbelkada

Parents