search

wuch15 / Fastformer

A pytorch &keras implementation and demo of Fastformer.
188 stars 29 forks source link

Reference to 'datasets' module #4

Closed Sheshansh closed 3 years ago

Sheshansh commented 3 years ago

The first line of Fastformer.ipynb is from datasets import load_dataset. But there is no datasets.py in the repository. Can you please share where I can find the datasets module/ script?

wuch15 commented 3 years ago

You can try pip install datasets. Note that this is the huggingface transformers API.

Sheshansh commented 3 years ago

Thanks a lot! This works!