Method __init__

losses/loss.py:6–12

(
        self,
        eps: float = 1e-8,
        )

class StrandVAELoss(torch.nn.Module):
    def __init__(
        self,
        eps: float = 1e-8,
        ):
        super().__init__()

        self.eps = eps
       
    def forward(self, phase, gt_dict, pred_dict, latent_dict, hyperparams):
        gt_strands= gt_dict["strand_positions"]