wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-08-11 05:51:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed typos

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-05-15 13:12:44 +02:00
parent b5dc3912cb
commit af0b129b21
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/distribution_output.py
+3 -5
View File
@@ -247,8 +247,7 @@ class LowRankMultivariateNormalOutput(DistributionOutput):
self.sigma_minimum = sigma_minimum
self.args_dim = {"loc": dim, "cov_factor": dim * rank, "cov_diag": dim}
@classmethod
def domain_map(cls, loc, cov_factor, cov_diag):
def domain_map(self, loc, cov_factor, cov_diag):
diag_bias = (
self.inv_softplus(self.sigma_init ** 2) if self.sigma_init > 0.0 else 0.0
)
@@ -299,9 +298,8 @@ class MultivariateNormalOutput(DistributionOutput):
self.args_dim = {"loc": dim, "scale_tril": dim * dim}
self.dim = dim
@classmethod
def domain_map(cls, loc, scale):
d = len(loc)
def domain_map(self, loc, scale):
d = self.dim
device = scale.device
shape = scale.shape[:-1] + (d, d)