transformers
[OPT] Run test in lower precision on GPU
#17353
Merged

[OPT] Run test in lower precision on GPU #17353

patrickvonplaten merged 8 commits into main from fix_opt_only_half
patrickvonplaten
patrickvonplaten [OPT] Run test only in half precision
db27c996
patrickvonplaten patrickvonplaten requested a review from ArthurZucker ArthurZucker 4 years ago
patrickvonplaten patrickvonplaten requested a review from ydshieh ydshieh 4 years ago
patrickvonplaten up
77b2f6eb
HuggingFaceDocBuilderDev
patrickvonplaten up
a533e063
ArthurZucker
ArthurZucker approved these changes on 2022-05-19
patrickvonplaten up
cccd9077
patrickvonplaten up
6002a650
patrickvonplaten finish
f72ae0da
patrickvonplaten
patrickvonplaten commented on 2022-05-19
patrickvonplaten
patrickvonplaten commented on 2022-05-19
patrickvonplaten fix on GPU
3db3ac25
patrickvonplaten patrickvonplaten changed the title [OPT] Run test only in half precision [OPT] Run test in lower precision on GPU 4 years ago
ydshieh
ydshieh approved these changes on 2022-05-19
patrickvonplaten
patrickvonplaten commented on 2022-05-19
patrickvonplaten Update tests/models/opt/test_modeling_opt.py
8f1cc48b
sgugger
sgugger approved these changes on 2022-05-19
patrickvonplaten patrickvonplaten merged e8714c03 into main 4 years ago
patrickvonplaten patrickvonplaten deleted the fix_opt_only_half branch 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone