mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add missing device argument to inference.py dataloader

Browse Source
This commit is contained in:
Ross Wightman 2024-04-10 09:34:34 -07:00
parent c28ee2e904
commit 195d12fdff
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
inference.py
View File

@ -241,6 +241,7 @@ def main():
batch_size=args.batch_size,
use_prefetcher=True,
num_workers=workers,
device=device,
**data_config,
)