Skip to content

v1.12

Compare
Choose a tag to compare
@freud14 freud14 released this 16 Jul 21:48
· 54 commits to master since this release
  • Fix a bug when transfering the optimizer on another device caused by a new feature in PyTorch 1.12, i.e. the "capturable" parameter in Adam and AdamW.
  • Add utilitary functions for saving (save_random_states) and loading (load_random_states) Python, Numpy and Pytorch's (both CPU and GPU) random states. Furthermore, we also add the RandomStatesCheckpoint callback. This callback is now used in ModelBundle.