Skip to content
Snippets Groups Projects
Commit d1ef1930 authored by Neta Zmora's avatar Neta Zmora
Browse files

CPU support: correct the device used for pruning masks

When masks are loaded from a checkpoint file, they should use the
same device as the model.
parent 0edfb5a9
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment