add get_class_indexing exception info

pull/318/head
Yanjun Peng 5 years ago
parent 8674e0ad96
commit 950d75e376

@ -782,7 +782,7 @@ class Dataset:
"""
if self.input:
return self.input[0].get_class_indexing()
return None
raise NotImplementedError("Dataset {} has not supported api get_class_indexing yet.".format(type(self)))
def reset(self):
"""Reset the dataset for next epoch"""

Loading…
Cancel
Save