-
- Downloads
Added support for setting the PRNG seed (#269)
Added set_seed() to Distiller and added support for seeding the PRNG when setting --deterministic mode (prior to this change, the seed is always set to zero when running in deterministic mode. The PRNGs of Pytorch (CPU & Cuda devices), numpy and Python are set. Added support for ```--seed``` to classifier_compression.py.
Showing
- distiller/utils.py 20 additions, 7 deletionsdistiller/utils.py
- examples/classifier_compression/compress_classifier.py 7 additions, 7 deletionsexamples/classifier_compression/compress_classifier.py
- examples/classifier_compression/parser.py 2 additions, 0 deletionsexamples/classifier_compression/parser.py
Loading
Please register or sign in to comment