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

Bug fix: compression schedule configuration parsing

Used the wrong indentation when parsing RegularizationPolicy
parent 73df57bf
No related branches found
No related tags found
No related merge requests found
......@@ -82,7 +82,7 @@ def dict_config(model, optimizer, sched_dict):
assert instance_name in regularizers, "Regularizer {} was not defined in the list of regularizers".format(instance_name)
regularizer = regularizers[instance_name]
if args is None:
policy = distiller.RegularizationPolicy(regularizer)
policy = distiller.RegularizationPolicy(regularizer)
else:
policy = distiller.RegularizationPolicy(regularizer, **args)
......
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