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):
state = {'masks_dict': masks}
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.
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