From fbcd5803b84d660dd6f8b4a0cab04e2929134d0f Mon Sep 17 00:00:00 2001
From: NikkeDoy <niklas.siltala@gmail.com>
Date: Mon, 7 Apr 2025 02:14:06 +0300
Subject: [PATCH] :bug: | Fixed training on CPU and NVIDIA hardware.

---
 training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/training.py b/training.py
index 380f738..c050b9c 100644
--- a/training.py
+++ b/training.py
@@ -119,7 +119,7 @@ if args.discriminator is not None:
 
 # Loss
 criterion_g = nn.MSELoss()
-criterion_d = nn.BCELoss()
+criterion_d = nn.BCEWithLogitsLoss()
 
 # Optimizers
 optimizer_g = optim.Adam(generator.parameters(), lr=0.0001, betas=(0.5, 0.999))