PR #2542 Fix test_script.py on TPU v2/v3

Fix test_script.py on TPU v2/v3 #2542

muellerzr merged 7 commits into huggingface:main from vanbasten23:fixDataloaderTake3

fix replication

e848ea6a

Set generator on each thread. The test passed.

497d4d15

remove comments

bc30f7d3

vanbasten23 changed the title ~~[DO NOT REVIEW YET] Fix test_script.py on TPU v2/v3~~ Fix test_script.py on TPU v2/v3 2 years ago

vanbasten23 marked this pull request as ready for review 2 years ago

muellerzr commented on 2024-03-11

will-cromar approved these changes on 2024-03-11

fix up

000e52ba

fix format

34b23b15

muellerzr commented on 2024-03-11

fix comment

ea334bf5

muellerzr commented on 2024-03-11

not setting the dataloader.batch_sampler

c12cb6fb

vanbasten23 requested a review from

muellerzr 2 years ago

muellerzr approved these changes on 2024-03-13

muellerzr merged 02a8a9a3 into main 2 years ago

Reviewers

muellerzr

will-cromar

Assignees

No one assigned

Labels

None yet

Milestone

No milestone