modified: dsm/dsm_torch.py

autonlab · Dec 14, 2020 · b3ad9a9 · b3ad9a9
1 parent edc176b
commit b3ad9a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()