You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This commit was created on GitHub.com and signed with GitHub’s verified signature.
Checkpoint Loading Improvements
Safe checkpoint loading: both Backbone.from_checkpoint and
AudioClassifier.from_checkpoint now use
map_location="cpu" and weights_only=True.
Weights always deserialize to CPU regardless of the device they were saved on,
letting the caller decide where to move the model. weights_only=True
prevents arbitrary code execution during unpickling.
Removed unused padding_mask argument from
BackboneConstructor.forward and forward_with_pooling.
