search

SforAiDl / KD_Lib

A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.
https://kd-lib.readthedocs.io/
MIT License
609 stars 58 forks source link

Use mock data for unit tests #131

Closed NeelayS closed 2 years ago

NeelayS commented 2 years ago

Use mock data for testing the methods instead of MNIST which is currently being used.