diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 03:47:29 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-12 03:47:29 +0100 |
| commit | 566f457e1492e4468e3739392cfe410a3c66d0e9 (patch) | |
| tree | 006e74cd2020534e4129a37a772559b587a23d57 | |
| parent | 85b8aea622c973a5e1643b04c13d39719fefca0e (diff) | |
new checkpoint reader
| -rw-r--r-- | cli/app/search/live.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py index 14f1ad3..c2127ab 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -88,13 +88,13 @@ def normal_z_sample(shape=(BATCH_SIZE, Z_DIM,)): # -------------------------- disentangled = { - 'zoom': read_checkpoint(os.path.join(app_cfg.DISENTANGLED, 'zoom/model.ckpt'), 'walk')[0], - 'shiftx': read_checkpoint(os.path.join(app_cfg.DISENTANGLED, 'shiftx/model.ckpt'), 'walk')[0], - 'shifty': read_checkpoint(os.path.join(app_cfg.DISENTANGLED, 'shifty/model.ckpt'), 'walk')[0], - 'rotate2d': read_checkpoint(os.path.join(app_cfg.DISENTANGLED, 'rotate2d/model.ckpt'), 'walk')[0], - 'rotate3d': read_checkpoint(os.path.join(app_cfg.DISENTANGLED, 'rotate3d/model.ckpt'), 'walk')[0], + '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.DISENTANGLED, 'rotate2d/model.ckpt'), 'walk')[0] +disentangled_color = read_checkpoint(os.path.join(app_cfg.DIR_DISENTANGLED, 'rotate2d/model.ckpt'), 'walk')[0] disentangled['r'] = disentangled_color[:, 0] disentangled['g'] = disentangled_color[:, 1] disentangled['b'] = disentangled_color[:, 2] |