X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=lazy_linear.py;h=97530ef2829bbae5912990adce85c31bdb4d93d9;hb=d74d7be5abef26c78d014bd179f2c52f81aca65b;hp=b3d316520f8b8344af4d2c2e87a82ca23707af64;hpb=bbc4ab46de8cb2e05f83e35e15f3adfd23490089;p=pytorch.git

diff --git a/lazy_linear.py b/lazy_linear.py
index b3d3165..97530ef 100755
--- a/lazy_linear.py
+++ b/lazy_linear.py
@@ -1,5 +1,10 @@
 #!/usr/bin/env python
 
+# Any copyright is dedicated to the Public Domain.
+# https://creativecommons.org/publicdomain/zero/1.0/
+
+# Written by Francois Fleuret <francois@fleuret.org>
+
 from torch import nn, Tensor
 
 ######################################################################
@@ -7,7 +12,7 @@ from torch import nn, Tensor
 class LazyLinear(nn.Module):
 
     def __init__(self, out_dim, bias = True):
-        super(LazyLinear, self).__init__()
+        super().__init__()
         self.out_dim = out_dim
         self.bias = bias
         self.core = None
@@ -25,7 +30,7 @@ class LazyLinear(nn.Module):
 
     def named_parameters(self, memo=None, prefix=''):
         assert self.core is not None, 'Parameters not yet defined'
-        return super(LazyLinear, self).named_parameters(memo, prefix)
+        return super().named_parameters(memo, prefix)
 
 ######################################################################