diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 02:12:33 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 02:12:33 +0100 |
| commit | f01105b36e5a8acf75d6a00f66ad9287d79f7575 (patch) | |
| tree | f6a69d758bc2690b73a7c2a30cb68140c73ae336 /inversion/image_inversion.py | |
| parent | 12e015807c1db187895aeca495b77b367c543211 (diff) | |
test small sample size
Diffstat (limited to 'inversion/image_inversion.py')
| -rw-r--r-- | inversion/image_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 04c033b..76638f9 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -355,7 +355,7 @@ for image_batch, label_batch in image_gen: # Save target. sess.run(target.assign(image_batch)) if COND_GAN: - sess.run(label.assign(one_hot(label_batch))) + sess.run(label.assign(label_batch)) # Initialize encodings to random values. if params.pre_trained_latent: |