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 trust_remote_code argument to ReaderHfds

Browse Source
This commit is contained in:
Augustin Godinot 2024-11-08 15:50:55 +01:00 committed by Ross Wightman
parent eb94efb218
commit 7f0c1b1f30
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/data/readers/reader_hfds.py
View File

@ -38,6 +38,7 @@ class ReaderHfds(Reader):
input_key: str = 'image',
target_key: str = 'label',
download: bool = False,
trust_remote_code: bool = False
):
"""
"""
@ -48,6 +49,7 @@ class ReaderHfds(Reader):
name, # 'name' maps to path arg in hf datasets
split=split,
cache_dir=self.root, # timm doesn't expect hidden cache dir for datasets, specify a path
trust_remote_code=trust_remote_code
)
# leave decode for caller, plus we want easy access to original path names...
self.dataset = self.dataset.cast_column(input_key, datasets.Image(decode=False))