Introduce new DiscriminatorLoss class, Remove useless super call params

This commit is contained in:
henryruhs
2025-03-11 14:43:10 +01:00
parent 579d3ef51c
commit 14b9bccafe
10 changed files with 84 additions and 49 deletions
@@ -4,7 +4,7 @@ from torch import Tensor, nn
class EmbeddingConverter(nn.Module):
def __init__(self) -> None:
super(EmbeddingConverter, self).__init__()
super().__init__()
self.layers = self.create_layers()
self.leaky_relu = nn.LeakyReLU()