Spaces:
Runtime error
Runtime error
Update fairseq/fairseq/data/indexed_dataset.py
Browse files
fairseq/fairseq/data/indexed_dataset.py
CHANGED
@@ -118,7 +118,7 @@ _code_to_dtype = {
|
|
118 |
3: np.int16,
|
119 |
4: np.int32,
|
120 |
5: np.int64,
|
121 |
-
6: np.
|
122 |
7: np.double,
|
123 |
8: np.uint16,
|
124 |
9: np.uint32,
|
@@ -325,7 +325,7 @@ class IndexedDatasetBuilder:
|
|
325 |
np.int16: 2,
|
326 |
np.int32: 4,
|
327 |
np.int64: 8,
|
328 |
-
np.
|
329 |
np.double: 8,
|
330 |
}
|
331 |
|
|
|
118 |
3: np.int16,
|
119 |
4: np.int32,
|
120 |
5: np.int64,
|
121 |
+
6: np.float64,
|
122 |
7: np.double,
|
123 |
8: np.uint16,
|
124 |
9: np.uint32,
|
|
|
325 |
np.int16: 2,
|
326 |
np.int32: 4,
|
327 |
np.int64: 8,
|
328 |
+
np.float64: 4,
|
329 |
np.double: 8,
|
330 |
}
|
331 |
|