projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[pytorch.git]
/
lazy_linear.py
diff --git
a/lazy_linear.py
b/lazy_linear.py
index
2fa6a29
..
97530ef
100755
(executable)
--- a/
lazy_linear.py
+++ b/
lazy_linear.py
@@
-12,7
+12,7
@@
from torch import nn, Tensor
class LazyLinear(nn.Module):
def __init__(self, out_dim, bias = True):
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
self.out_dim = out_dim
self.bias = bias
self.core = None
@@
-30,7
+30,7
@@
class LazyLinear(nn.Module):
def named_parameters(self, memo=None, prefix=''):
assert self.core is not None, 'Parameters not yet defined'
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)
######################################################################
######################################################################