save all

2023-08-24 11:36:22 +00:00
parent 59f67a2495
commit afd482522d
36 changed files with 2194 additions and 291 deletions
@@ -39,7 +39,7 @@ class PyramidLevel(nn.Module):

    def sample(self, z_cond, uv_mask, noise=None):
        if noise is None:
-            eps = torch.randn(z_cond.shape).to(uv_mask.device)
+            eps = torch.normal(mean=0.0, std=1.0, size=z_cond.shape).to(uv_mask.device)
        else:
            eps = noise