Skip to content
Snippets Groups Projects
Commit a0b02c1c authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[trainer] small synchro

parent ec3dfa34
Branches
Tags
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment