genie-toolkit
genie-toolkit copied to clipboard
Augmentation is memory hungry
Currently, the augmentation step will load all parameter values to memory, which could be an issue for some cases like QA over Wikidata. We could either use mmap or do a pre-sampling over the dataset.