-
- Downloads
MNIST support
-Added a test for MNIST -Added classification_get_dummy_input() to apputils/data_loaders.py and wrapped it with get_dummy_input() for (temporary) backward compatibility. - Changed simplenet_mnist so that it supports thinning
Showing
- distiller/apputils/data_loaders.py 18 additions, 0 deletionsdistiller/apputils/data_loaders.py
- distiller/models/mnist/simplenet_mnist.py 1 addition, 1 deletiondistiller/models/mnist/simplenet_mnist.py
- distiller/utils.py 2 additions, 15 deletionsdistiller/utils.py
- tests/test_model_summary.py 8 additions, 1 deletiontests/test_model_summary.py
Loading
Please register or sign in to comment