xla
[torchbench] Re-land: Force `llama_v2_7b_16h` to use SGD.
#6402

Merged

[torchbench] Re-land: Force `llama_v2_7b_16h` to use SGD. #6402

ysiraichi merged 1 commit into master from ysiraichi/fix-llama2-optimizer

Force `llama_v2_7b_16h` to use SGD.

57a1a6bb

ysiraichi added xla:gpu

ysiraichi requested a review from

zpcore 1 year ago

ysiraichi requested a review from

frgossen 1 year ago

ysiraichi requested a review from

vanbasten23 1 year ago

ysiraichi requested a review from

golechwierowicz 1 year ago

zpcore approved these changes on 2024-01-30

ysiraichi merged 0baf1dae into master 1 year ago

Reviewers

zpcore

frgossen

vanbasten23

golechwierowicz

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone