examples/classifier_compression/inspect_ckpt.py · 8c5de42cf08e8d32b966a570c207039cb799a528 · llvm / distiller

6 years ago

save_checkpoint(): Save dict of user values (Breaking) (#219) · 158602c5

Guy Jacob authored 6 years ago

* 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'

Unverified

158602c5

History

save_checkpoint(): Save dict of user values (Breaking) (#219)

Guy Jacob authored 6 years ago

* 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'

inspect_ckpt.py 3.06 KiB