Version 0.15.0
This is a smaller release, but it still contains changes which will be interesting to some of you.
We added the possibility to store weights using safetensors. This can have several advantages, listed here. When calling net.save_params
and net.load_params
, just pass use_safetensors=True
to use safetensors instead of pickle.
Moreover, there is a new argument on NeuralNet
: You can now pass use_caching=False
or True
to disable or enable caching for all callbacks at once. This is useful if you have a lot of scoring callbacks and don't want to toggle caching on each individually.
Finally, we fixed a few issues related to using skorch with accelerate.
Thanks Zach Mueller (@muellerzr) for his first contribution to skorch.
Find the full list of changes here: v0.14.0...v0.15.0