We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent eb94efb commit 7f0c1b1Copy full SHA for 7f0c1b1
timm/data/readers/reader_hfds.py
@@ -38,6 +38,7 @@ def __init__(
38
input_key: str = 'image',
39
target_key: str = 'label',
40
download: bool = False,
41
+ trust_remote_code: bool = False
42
):
43
"""
44
@@ -48,6 +49,7 @@ def __init__(
48
49
name, # 'name' maps to path arg in hf datasets
50
split=split,
51
cache_dir=self.root, # timm doesn't expect hidden cache dir for datasets, specify a path
52
+ trust_remote_code=trust_remote_code
53
)
54
# leave decode for caller, plus we want easy access to original path names...
55
self.dataset = self.dataset.cast_column(input_key, datasets.Image(decode=False))
0 commit comments