check signing my commit

dice-group · Dec 17, 2024 · 741b129 · 741b129
1 parent 2df1d30
commit 741b129
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/dicee/models/base_model.py b/dicee/models/base_model.py
@@ -136,14 +136,6 @@ def __init__(self, args: dict):
         self.kernel_size = None
         self.num_of_output_channels = None
         self.weight_decay = None
-
-        if self.args["loss_fn"] == "BCELoss":
-            self.loss = torch.nn.BCEWithLogitsLoss()
-        if self.args["loss_fn"] == "LRLoss":
-            self.loss = LabelRelaxationLoss()
-        else:
-            self.loss = torch.nn.BCEWithLogitsLoss()
-
         self.selected_optimizer = None
         self.normalizer_class = None
         self.normalize_head_entity_embeddings = IdentityClass()
@@ -162,6 +154,14 @@ def __init__(self, args: dict):
         self.byte_pair_encoding = self.args.get("byte_pair_encoding", False)
         self.max_length_subword_tokens = self.args.get("max_length_subword_tokens", None)
         self.block_size=self.args.get("block_size", None)
+
+        if self.args["loss_fn"] == "BCELoss":
+            self.loss = torch.nn.BCEWithLogitsLoss()
+        if self.args["loss_fn"] == "LRLoss":
+            self.loss = LabelRelaxationLoss()
+        else:
+            self.loss = torch.nn.BCEWithLogitsLoss()
+
         if self.byte_pair_encoding and self.args['model'] != "BytE":
             self.token_embeddings = torch.nn.Embedding(self.num_tokens, self.embedding_dim)
             self.param_init(self.token_embeddings.weight.data)