From 7e89a45492007c8a4c758ebfd0fe0b19a313c825 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Simon=20Ekstr=C3=B6m?= Date: Thu, 4 Jul 2019 12:39:03 +0200 Subject: [PATCH] Fixed small error in variable name --- register.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/register.py b/register.py index dc9c39e..ea89d97 100644 --- a/register.py +++ b/register.py @@ -91,7 +91,7 @@ def __init__(self, dim): def add_initial_transform(self, transform, param_scaling=None): if param_scaling is None: - param_scaling = np.ones((transforms.get_param_count(),)) + param_scaling = np.ones((transform.get_param_count(),)) self.initial_transforms.append(transform) self.transforms_param_scaling.append(param_scaling)