Skip to content

Dataprovider iterator

Thibault Allenet requested to merge dataproviderIterator into dev

Add features to dataprovider :

  • Pybind iterator for dataProvider with __iter__, __next__ and __len__ implemented functions
  • dataProvider shuffle argument to perform random batch permutation for learning
  • dataProvider dropLast argument to drop the last non-full batch

Other :

  • Add random utils to perform random batch permutation for learning
  • Add support for Int8, Int16, Uint8, Uint16 for tensor python binding
Edited by Thibault Allenet

Merge request reports

Loading