From 575f2154085acb2f0e8e056d3eb1754a94ab942f Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 21 Feb 2025 18:40:11 +0100 Subject: [PATCH] lr only in optimizer --- embedding_converter/src/training.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index 3d66360..b323fe2 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -24,7 +24,6 @@ class EmbeddingConverterTrainer(lightning.LightningModule): super(EmbeddingConverterTrainer, self).__init__() source_path = CONFIG.get('training.model', 'source_path') target_path = CONFIG.get('training.model', 'target_path') - learning_rate = CONFIG.getfloat('training.trainer', 'learning_rate') self.embedding_converter = EmbeddingConverter() self.source_embedder = torch.jit.load(source_path) # type:ignore[no-untyped-call] @@ -32,7 +31,6 @@ class EmbeddingConverterTrainer(lightning.LightningModule): self.source_embedder.eval() self.target_embedder.eval() self.mse_loss = nn.MSELoss() - self.lr = learning_rate def forward(self, source_embedding : Embedding) -> Embedding: return self.embedding_converter(source_embedding)