xla
Set gradient_as_bucket_view=True in test_train_mp_imagenet
#8558

Merged

Set gradient_as_bucket_view=True in test_train_mp_imagenet #8558

tengyifei merged 2 commits into master from yifeit/gradient-as-view-resnet

Set gradient_as_bucket_view=True in test_train_mp_imagenet

2bcd74af

tengyifei requested a review from

yaochengji 259 days ago

tengyifei marked this pull request as ready for review 259 days ago

yaochengji approved these changes on 2025-01-13

tengyifei requested a review from

mikegre-google 259 days ago

Add back gradient_as_bucket_view=True in our examples

3fbd6958

tengyifei force pushed from 558d71de to 3fbd6958 259 days ago

mikegre-google approved these changes on 2025-01-13

yaochengji approved these changes on 2025-01-13

tengyifei merged 2fdf721f into master 259 days ago

Reviewers

yaochengji

mikegre-google

Assignees

No one assigned

Labels

None yet

Milestone

No milestone