Skip to content

GitLab

Unverified Commit 683baf8e authored by Adam J. Stewart's avatar Adam J. Stewart Committed by GitHub
Browse files

Check sha256 of weights (#7219) 


Co-authored-by: default avatarNicolas Hug <nh.nicolas.hug@gmail.com>
parent 8811c915
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/prototype/models/depth/stereo/raft_stereo.py
View file @ 683baf8e
......@@ -620,7 +620,7 @@ def _raft_stereo(
)
if weights is not None:
model.load_state_dict(weights.get_state_dict(progress=progress))
model.load_state_dict(weights.get_state_dict(progress=progress, check_hash=True))
return model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment