Make EPSILON a tensor

qarac/models/QaracTrainerModel.py

@@ -10,7 +10,7 @@ import torch
 import qarac.models.QaracEncoderModel
 import qarac.models.QaracDecoderModel
 
-EPSILON=1.0e-12
+EPSILON=torch.tensor(1.0e-12)
 
 class QaracTrainerModel(torch.nn.Module):
     

qarac/models/layers/GlobalAttentionPoolingHead.py

@@ -8,7 +8,7 @@ Created on Tue Sep  5 07:32:55 2023
 
 import torch
 
-EPSILON = 1.0e-12   
+EPSILON = torch.tensor(1.0e-12)   
 
 class GlobalAttentionPoolingHead(torch.nn.Module):
     

scripts.py

@@ -22,7 +22,7 @@ import scipy.spatial
 import seaborn
 import tqdm
 
-EPSILON = 1.0e-12
+EPSILON = torch.tensor(1.0e-12)
 
 class CombinedLoss(torch.nn.Module):
     def __init__(self):