black formatted files before changes

This commit is contained in:
Jan Kowalczyk
2024-06-28 11:36:46 +02:00
parent d33c6b1e16
commit 71f9662022
40 changed files with 2938 additions and 1260 deletions

View File

@@ -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
)