-
- Downloads
save_checkpoint(): Save dict of user values (Breaking) (#219)
* Replace the optional 'best_top1' parameter with a generic optional dict which the caller can populate as needed. * Saved in the checkpoint under the key 'extras'
Showing
- distiller/apputils/checkpoint.py 33 additions, 12 deletionsdistiller/apputils/checkpoint.py
- distiller/pruning/greedy_filter_pruning.py 2 additions, 2 deletionsdistiller/pruning/greedy_filter_pruning.py
- examples/classifier_compression/compress_classifier.py 11 additions, 6 deletionsexamples/classifier_compression/compress_classifier.py
- examples/classifier_compression/inspect_ckpt.py 9 additions, 16 deletionsexamples/classifier_compression/inspect_ckpt.py
Loading
Please register or sign in to comment