diff --git a/distiller/config.py b/distiller/config.py index a5cc9cd76a82b7c90e873f9d27a7cafc5244742c..0e2fd628f0650a5694e2b0ff2b368bb52de5afeb 100755 --- a/distiller/config.py +++ b/distiller/config.py @@ -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)