fix perceptual loss

nnaakkaaii · Jul 1, 2024 · 49ef9b1 · 49ef9b1
1 parent cb52347
commit 49ef9b1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/hrdae/conf b/hrdae/conf
diff --git a/hrdae/models/losses/perceptual.py b/hrdae/models/losses/perceptual.py
@@ -44,6 +44,7 @@ def __init__(
         if torch.cuda.is_available():
             self.network.to("cuda:0")
             self.network = nn.DataParallel(self.network)
+        self.network.eval()
 
     def forward(self, input: Tensor, target: Tensor) -> Tensor:
         b, t = input.size()[:2]