search

JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡
https://julesbelveze.github.io/bert-squeeze/
79 stars 10 forks source link

Wrong path to dataset in `DataModule` #22

Closed JulesBelveze closed 2 years ago

JulesBelveze commented 2 years ago

The load_dataset method doesn't load local datasets from the correct folder.

"data/{self.dataset_config.name}_dataset.py"

needs to be changed to 

"data/datasets/{self.dataset_config.name}_dataset.py"