NikkeDoy/SISU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 | Fixed training on CPU and NVIDIA hardware.

Browse Source
This commit is contained in:
NikkeDoy 2025-04-07 02:14:06 +03:00
parent 9394bc6c5a
commit fbcd5803b8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
training.py
View File

@ -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))