def load_wikitext()

in opacus_lab/models/GPT2/dataset.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

def load_wikitext(path):
    corpus = dict()
    for dset in ["valid", "train", "test"]:
        corpus[dset] = torch.load(f"{path}/wikitext-103-{dset}-corpus.pt")
    return corpus