summaryrefslogtreecommitdiff
path: root/inversion/live.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-12-19 19:57:34 +0100
committerJules Laplace <julescarbon@gmail.com>2019-12-19 19:57:34 +0100
commitcf18722d3620db3c10972299f7039049a017a738 (patch)
tree3f75e81d2207550403c2eafd6e29ec0af379e78f /inversion/live.py
parentd0e9621658be1e159f2ac87dc6ebf80ba4b7caec (diff)
redo lerping and sineing
Diffstat (limited to 'inversion/live.py')
-rw-r--r--inversion/live.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/inversion/live.py b/inversion/live.py
index 08cc789..b3e41b1 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -85,7 +85,7 @@ def normal_z_sample(shape=(BATCH_SIZE, Z_DIM,)):
# --------------------------
class SinParam:
- def __init__(self, name, shape, type):
+ def __init__(self, name, shape, type="noise"):
noise = LerpParam(name + '_noise', shape, type=type)
orbit_radius = InterpolatorParam(name=name + '_radius', value=0.1)
orbit_speed = InterpolatorParam(name=name + '_speed', value=0.1)
@@ -168,7 +168,7 @@ class Interpolator:
self.opts['num_classes'] = InterpolatorParam(name='num_classes', value=1.0),
lerp_z = LerpParam('latent', [BATCH_SIZE, Z_DIM])
sin_z = SinParam('orbit', [BATCH_SIZE, Z_DIM])
- lerp_label = LerpParam('label', [BATCH_SIZE, N_CLASS])
+ lerp_label = LerpParam('label', [BATCH_SIZE, N_CLASS], type="label")
# self.opts['z'] = InterpolatorParam(name='z', shape=[BATCH_SIZE, Z_DIM], type='noise')
# self.opts['y'] = InterpolatorParam(name='y', shape=[BATCH_SIZE, N_CLASS], type='label')