diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 20:17:26 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 20:17:26 +0100 |
| commit | dda80e68dce16622cf8ddc0fe197615a4be6f7b4 (patch) | |
| tree | 29f455f66c5a33d714f203e7726d69bd3b7e69ae /cli/app/search/live.py | |
| parent | d32a222b685ea5b4f340bc6706dbb761fe540cb4 (diff) | |
fix dims
Diffstat (limited to 'cli/app/search/live.py')
| -rw-r--r-- | cli/app/search/live.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py index f066faf..ca54a76 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -266,11 +266,11 @@ class Interpolator: def load_disentangled_latents(self): self.disentangled = { - 'zoom': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'zoom/model.ckpt'), 'walk'), - 'shiftx': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'shiftx/model.ckpt'), 'walk'), - 'shifty': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'shifty/model.ckpt'), 'walk'), - 'rotate2d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate2d/model.ckpt'), 'walk'), - 'rotate3d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate3d/model.ckpt'), 'walk'), + 'zoom': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'zoom/model.ckpt'), 'walk')[:, :, 0], + 'shiftx': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'shiftx/model.ckpt'), 'walk')[:, :, 0], + 'shifty': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'shifty/model.ckpt'), 'walk')[:, :, 0], + 'rotate2d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate2d/model.ckpt'), 'walk')[:, :, 0], + 'rotate3d': read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate3d/model.ckpt'), 'walk')[:, :, 0], } disentangled_color = read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'color/model.ckpt'), 'walk') self.disentangled['r'] = disentangled_color[:, :, 0] |