Skip to content
Snippets Groups Projects
Unverified Commit fe27ab90 authored by Neta Zmora's avatar Neta Zmora Committed by GitHub
Browse files

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.
parent 5a0f967b
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment