From 257ab668eec2ad8144d668141952dfdf1716a15e Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 13 Feb 2025 19:02:00 +0100 Subject: [PATCH] Name what it is --- embedding_converter/src/models/embedding_converter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/embedding_converter/src/models/embedding_converter.py b/embedding_converter/src/models/embedding_converter.py index 9485779..d9c61c4 100644 --- a/embedding_converter/src/models/embedding_converter.py +++ b/embedding_converter/src/models/embedding_converter.py @@ -8,7 +8,7 @@ class EmbeddingConverter(nn.Module): def __init__(self) -> None: super(EmbeddingConverter, self).__init__() self.layers = self.create_layers() - self.activation = nn.LeakyReLU() + self.leaky_relu = nn.LeakyReLU() @staticmethod def create_layers() -> nn.ModuleList: @@ -24,7 +24,7 @@ class EmbeddingConverter(nn.Module): output_tensor = input_tensor / torch.norm(input_tensor) for layer in self.layers[:-1]: - output_tensor = self.activation(layer(output_tensor)) + output_tensor = self.leaky_relu(layer(output_tensor)) output_tensor = self.layers[-1](output_tensor) return output_tensor