Merge branch 'master' into indexer-merge

7a348580 · Neel Kant · b9bd1a11 · 05620ee4 · 7a348580
Commit 7a348580 authored Jul 14, 2020 by Neel Kant
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

megatron/data/gpt2_dataset.py megatron/data/gpt2_dataset.py +1 -1

No files found.
--- a/megatron/data/gpt2_dataset.py
+++ b/megatron/data/gpt2_dataset.py
@@ -22,7 +22,7 @@ import numpy as np
 import torch

 from megatron import mpu, print_rank_0
-from megatron.data.bert_dataset import get_train_valid_test_split_
+from megatron.data.dataset_utils import get_train_valid_test_split_
 from megatron.data.indexed_dataset import make_dataset as make_indexed_dataset