diff options
| author | Vee9ahd1 <[email protected]> | 2019-09-22 19:43:03 -0400 |
|---|---|---|
| committer | Christian <[email protected]> | 2019-09-22 19:46:11 -0400 |
| commit | 044232a0947fcf02118a268e45a615cc1a8065a2 (patch) | |
| tree | 213e6375107eee8cb5cf66dc4fbdc661200f0cb1 /gantools/latent_space.py | |
| parent | ed90828eb9712de21ecf7bfa5dfe88bd27672a0b (diff) | |
patched for artbreeder support and fixed some weird CUDNN_STATUS_INTERNAL_ERROR. if CPU support is broken now, this commit is probably why.
Diffstat (limited to 'gantools/latent_space.py')
| -rw-r--r-- | gantools/latent_space.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/gantools/latent_space.py b/gantools/latent_space.py index 64e3dec..a39e804 100644 --- a/gantools/latent_space.py +++ b/gantools/latent_space.py @@ -44,7 +44,7 @@ def sequence_keyframes(keyframes, num_frames, batch_size=1, interp_method='linea keyframes.append(keyframes[0])# seq returns to start truncation_keys = np.asarray([keyframe['truncation'] for keyframe in keyframes]) - z_keys = np.asarray([np.asarray(keyframe['vector']) * keyframe['truncation'] for keyframe in keyframes]) + z_keys = np.asarray([np.asarray(keyframe['latent']) * keyframe['truncation'] for keyframe in keyframes]) label_keys = np.asarray([keyframe['label'] for keyframe in keyframes]) z_seq = interp_fn(z_keys, num_frames) |