fix: correct typo

2022-04-04 21:50:22 +02:00
parent 71a2e74eff
commit 0d10fc7e25
2 changed files with 6 additions and 6 deletions
--- a/prototorch/core/transforms.py
+++ b/prototorch/core/transforms.py
@@ -5,7 +5,7 @@ from torch.nn.parameter import Parameter

 from .initializers import (
    AbstractLinearTransformInitializer,
-    EyeTransformInitializer,
+    EyeLinearTransformInitializer,
 )


@@ -15,7 +15,7 @@ class LinearTransform(torch.nn.Module):
        in_dim: int,
        out_dim: int,
        initializer:
-        AbstractLinearTransformInitializer = EyeTransformInitializer()):
+        AbstractLinearTransformInitializer = EyeLinearTransformInitializer()):
        super().__init__()
        self.set_weights(in_dim, out_dim, initializer)

@@ -31,7 +31,7 @@ class LinearTransform(torch.nn.Module):
        in_dim: int,
        out_dim: int,
        initializer:
-        AbstractLinearTransformInitializer = EyeTransformInitializer()):
+        AbstractLinearTransformInitializer = EyeLinearTransformInitializer()):
        weights = initializer.generate(in_dim, out_dim)
        self._register_weights(weights)