xla
Use SGD for list of models from PyTorch.
#6324
Merged

Use SGD for list of models from PyTorch. #6324

ysiraichi merged 4 commits into master from ysiraichi/fix-model-optimizer
ysiraichi
ysiraichi ysiraichi requested a review from frgossen frgossen 2 years ago
ysiraichi ysiraichi requested a review from golechwierowicz golechwierowicz 2 years ago
ysiraichi ysiraichi requested a review from cota cota 2 years ago
ysiraichi ysiraichi requested a review from vanbasten23 vanbasten23 2 years ago
vanbasten23
vanbasten23 approved these changes on 2024-01-19
golechwierowicz
golechwierowicz commented on 2024-01-19
ysiraichi Use SGD optimizer for llama model.
d928f9d7
ysiraichi Extract list from PyTorch.
9dc00b92
ysiraichi Fix lint issues.
f0620510
ysiraichi Use `BENCHMARK_USE_SGD` set, instead.
b6d6acd0
ysiraichi ysiraichi force pushed from cfae2387 to b6d6acd0 2 years ago
ysiraichi ysiraichi merged a7918a72 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone