Change dataloader send_to_device calls to non-blocking #2685
muellerzr
merged
8 commits into
huggingface:main
from
drhead:patch-1
Change dataloader send_to_device calls to non-blocking
390c5377
add non_blocking to dataloader dataclass
b6a9cfa5
add dataloader non blocking option from dataclass
6eccc733
add handling for non blocking to accelerator
8b345f4e
add notes on non-blocking transfers to quicktour
975449d2
link to dataloaderconfiguration in docs
1d202880
muellerzr
approved these changes
on 2024-04-24
linting
4ec38729
"requires" -> "recommended" on non-blocking setting
0a821810
muellerzr
approved these changes
on 2024-04-24
Login to write a write a comment.
Login via GitHub