diff options
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 af8568f..5556778 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -348,7 +348,7 @@ out_enc = out_file.create_dataset('encoding', [NUM_IMGS,] + ENC_SHAPE) out_lat = out_file.create_dataset('latent', [NUM_IMGS, Z_DIM]) out_fns = out_file.create_dataset('fn', [NUM_IMGS], dtype=h5py.string_dtype()) if COND_GAN: - out_labels = out_file.create_dataset('ytrain', (NUM_IMGS,), dtype='uint32') + out_labels = out_file.create_dataset('ytrain', (NUM_IMGS, N_CLASS,), dtype='uint32') out_err = out_file.create_dataset('err', (NUM_IMGS,)) # Gradient descent w.r.t. generator's inputs. |