How to train my original dataset in distiller?

IntelLabs / distiller

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

Apache License 2.0

4.34k stars 799 forks source link

How to train my original dataset in distiller? #562

Open priNs0123 opened 3 years ago

priNs0123 commented 3 years ago

Does anyone know how to apply my original dataset to distiller? It seems that Distiller only supports very basic dataset like MNIST, cifar10 and Imagenet.

TigerVersusT commented 2 years ago

simple, just modify dataloading codes by your own, because there seems no APIs about extending user-defined datasets