Fix Loss to device

MALES-project · Mar 17, 2024 · 9abc481 · 9abc481
1 parent ef478f1
commit 9abc481
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/speckcn2/loss.py b/speckcn2/loss.py
@@ -61,13 +61,6 @@ def __init__(self, config: dict, nz: Normalizer, device: torch.device):
         self.recover_tag = nz.recover_tag
         # Move tensors to the device
         self.h = self.h.to(self.device)
-        self.k = self.k.to(self.device)
-        self.cosz = torch.tensor(self.cosz, device=self.device)
-        self.secz = torch.tensor(self.secz, device=self.device)
-        self.L = torch.tensor(self.L, device=self.device)
-        self.p_fr = torch.tensor(self.p_fr, device=self.device)
-        self.p_iso = torch.tensor(self.p_iso, device=self.device)
-        self.p_scw = torch.tensor(self.p_scw, device=self.device)
 
     def forward(self, pred: torch.Tensor,
                 target: torch.Tensor) -> tuple[torch.Tensor, dict]: