-
Notifications
You must be signed in to change notification settings - Fork 214
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
support using cached data and re-splitting for huggingface datasets #302
Conversation
…the GLUE versions into custom FL versions
…the GLUE versions into custom FL versions
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Good job, we could accelerate to load the huggingface data file with the cache.
tokenizer = AutoTokenizer.from_pretrained( | ||
config.model.type.split('@')[0]) | ||
|
||
try: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why need a try here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
In case of the cached file not existed
@@ -402,6 +413,7 @@ def load_torch_geometric_data(name, splits=None, config=None): | |||
|
|||
def load_huggingface_datasets_data(name, splits=None, config=None): | |||
from datasets import load_dataset | |||
from datasets import load_from_disk |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Merge Line 415 and Line 416.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
support