Spaces:
Running
Running
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 |
|