From 6fcd7d24037abdc2d12b856f8ae50dece6cc85ca Mon Sep 17 00:00:00 2001 From: "askerosted@gmail.com" Date: Tue, 21 May 2024 10:57:13 +0900 Subject: [PATCH] unused params fix --- src/graphnet/models/components/embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/graphnet/models/components/embedding.py b/src/graphnet/models/components/embedding.py index e97ca90e7..1b49cd901 100644 --- a/src/graphnet/models/components/embedding.py +++ b/src/graphnet/models/components/embedding.py @@ -84,7 +84,6 @@ def __init__( super().__init__() self.sin_emb = SinusoidalPosEmb(dim=seq_length, scaled=scaled) - self.aux_emb = nn.Embedding(2, seq_length // 2) self.sin_emb2 = SinusoidalPosEmb(dim=seq_length // 2, scaled=scaled) if n_features < 4: @@ -93,6 +92,7 @@ def __init__( f"{n_features} features." ) elif n_features >= 6: + self.aux_emb = nn.Embedding(2, seq_length // 2) hidden_dim = 6 * seq_length else: hidden_dim = int((n_features + 0.5) * seq_length)