Commit a0b02c1c authored by Guillaume HEUSCH's avatar Guillaume HEUSCH

[trainer] small synchro

parent ec3dfa34
......@@ -119,14 +119,15 @@ class ConditionalGANTrainer(object):
start = time.time()
# ===========================================================
# (1) Update D network: maximize log(D(x)) + log(1 - D(G(z)))
# ===========================================================
# =============
# DISCRIMINATOR
# =============
# train with real
self.netD.zero_grad()
real_images = data['image']
poses = data['pose']
image_size = real_images[1].size()
# WARNING: the last batch could be smaller than the provided size
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment