lots of models, inc NP

2026-09-11 12:44:11 +08:00 · 2020-10-19 20:51:23 +08:00
parent e986751e41
commit 7b6c729db5
12 changed files with 3554 additions and 1623 deletions
@@ -76,5 +76,5 @@ class TransformerSeq2Seq(nn.Module):
        mean = self.mean(outputs)
        log_sigma = self.std(outputs)
        sigma = self._min_std + (1 - self._min_std) * F.softplus(log_sigma)
-        return torch.distributions.Normal(mean, sigma)
+        return torch.distributions.Normal(mean, sigma), {}