Skip to content

Commit 88fbdc2

Browse files
author
Leander Lauenburg
committed
fix: correct Ds loss wiring and scheduler hookup
1 parent 9063b27 commit 88fbdc2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

projects/CySGAN/cysgan/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def train(self):
170170
real_seg = torch.cat(targetX, 1).to(self.device) # concatenate over channel dim
171171
fake_seg = self.seg_handler(fakeXseg if random.random() > 0.5 else recYseg)
172172
loss_Ds = self.update_netD(self.Ds, real_seg, self.image_pool['Ds'].query(fake_seg),
173-
self.optimizer['Dy'], self.lr_scheduler['Dy'])
173+
self.optimizer['Ds'], self.lr_scheduler['Ds'])
174174
loss_D = loss_Dx + loss_Dy + loss_Ds # discriminator losses
175175

176176
self.iter_time = time.perf_counter() - self.start_time

0 commit comments

Comments
 (0)