Add support for different TFDS `BuilderConfig`s

3 years ago · c76d772670
parent 7eeaf521a0
commit c76d772670
1 changed files with 1 additions and 1 deletions
--- a/timm/data/parsers/parser_factory.py
+++ b/timm/data/parsers/parser_factory.py
@ -11,7 +11,7 @@ def create_parser(name, root, split='train', **kwargs):
    prefix = ''
    if len(name) > 1:
        prefix = name[0]
-    name = name[-1]
+    name = "/".join(name[1:])
    # FIXME improve the selection right now just tfds prefix or fallback path, will need options to
    # explicitly select other options shortly