diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 10:12:32 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 10:12:32 +0100 |
| commit | 7aad285136f97d737ef251ae35ed77404a847bb7 (patch) | |
| tree | ddaf4a86c137ec124b30a72ed0c2cb4456645a23 /inversion/image_inversion.py | |
| parent | 44f07d8dbbdace759150169cca0fafcaab8e1dd5 (diff) | |
store classes
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. |