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

Merge new-arch, because it has proven to give the best results #1

Merged
NikkeDoy merged 14 commits from new-arch into main 2025-04-30 23:47:41 +03:00
Conversation 0 Commits 14 Files Changed 8 +1 -1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit fbcd5803b8 - Show all commits

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