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

scheduler: backward compatibility for the signature of load_state_dict

Add a default value for the new parameter 'normalize_dataparallel_keys'
parent 42b0e695
No related branches found
No related tags found
No related merge requests found
...@@ -189,7 +189,7 @@ class CompressionScheduler(object): ...@@ -189,7 +189,7 @@ class CompressionScheduler(object):
state = {'masks_dict': masks} state = {'masks_dict': masks}
return state return state
def load_state_dict(self, state, normalize_dataparallel_keys): def load_state_dict(self, state, normalize_dataparallel_keys=False):
"""Loads the scheduler state. """Loads the scheduler state.
Currently the scheduler state is comprised only of the set of pruning masks. Currently the scheduler state is comprised only of the set of pruning masks.
......
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