black formatted files before changes
This commit is contained in:
@@ -21,7 +21,8 @@ class Standardize(Module):
|
||||
mu: the learnable translation parameter μ.
|
||||
std: the learnable scale parameter σ.
|
||||
"""
|
||||
__constants__ = ['mu']
|
||||
|
||||
__constants__ = ["mu"]
|
||||
|
||||
def __init__(self, in_features, bias=True, eps=1e-6):
|
||||
super(Standardize, self).__init__()
|
||||
@@ -32,7 +33,7 @@ class Standardize(Module):
|
||||
if bias:
|
||||
self.mu = Parameter(torch.Tensor(in_features))
|
||||
else:
|
||||
self.register_parameter('mu', None)
|
||||
self.register_parameter("mu", None)
|
||||
self.reset_parameters()
|
||||
|
||||
def reset_parameters(self):
|
||||
@@ -47,6 +48,6 @@ class Standardize(Module):
|
||||
return x
|
||||
|
||||
def extra_repr(self):
|
||||
return 'in_features={}, out_features={}, bias={}'.format(
|
||||
return "in_features={}, out_features={}, bias={}".format(
|
||||
self.in_features, self.out_features, self.mu is not None
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user