From b3ad9a9223b980df42b34dfc42a5d713d84009c4 Mon Sep 17 00:00:00 2001 From: Chirag Nagpal Date: Mon, 14 Dec 2020 21:42:39 +0530 Subject: [PATCH] modified: dsm/dsm_torch.py --- dsm/dsm_torch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsm/dsm_torch.py b/dsm/dsm_torch.py index 551a875..23bce8d 100644 --- a/dsm/dsm_torch.py +++ b/dsm/dsm_torch.py @@ -152,7 +152,7 @@ def __init__(self, inputdim, k, layers=None, dist='Weibull', self.act = nn.SELU() self.shape = nn.ParameterDict({str(r+1): nn.Parameter(-torch.ones(k)) for r in range(self.risks)}) - self.scale = nn.ParameterDict({str(r+1):nn.Parameter(-torch.ones(k)) + self.scale = nn.ParameterDict({str(r+1): nn.Parameter(-torch.ones(k)) for r in range(self.risks)}) elif self.dist in ['Normal']: self.act = nn.Identity()