diff --git a/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py b/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py index b4febcd822e1..e605f608054c 100644 --- a/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py +++ b/examples/tutorial/sequence_parallel/data/datasets/indexed_dataset.py @@ -94,7 +94,7 @@ def write_longs(f, a): 3: np.int16, 4: np.int32, 5: np.int64, - 6: np.float, + 6: np.float64, 7: np.double, 8: np.uint16 } @@ -267,7 +267,7 @@ class IndexedDatasetBuilder(object): np.int16: 2, np.int32: 4, np.int64: 8, - np.float: 4, + np.float64: 4, np.double: 8 } diff --git a/examples/tutorial/sequence_parallel/requirements.txt b/examples/tutorial/sequence_parallel/requirements.txt index b49a94554afb..4fc576453de8 100644 --- a/examples/tutorial/sequence_parallel/requirements.txt +++ b/examples/tutorial/sequence_parallel/requirements.txt @@ -1,2 +1,3 @@ colossalai torch +six