DtYXs commited on
Commit
7116e49
1 Parent(s): d55c6e0

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.float,
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.float: 4,
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