resolve conflict

nnaakkaaii · Jul 1, 2024 · 626d94d · 626d94d
2 parents 741c9eb + 49ef9b1
commit 626d94d
Showing 1 changed file with 1 addition and 0 deletions.
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]