speed param

1 files changed, 7 insertions, 3 deletions

diff --git a/inversion/live.py b/inversion/live.py
index 23af747..50ac25f 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -88,7 +88,7 @@ def normal_z_sample(shape=(BATCH_SIZE, Z_DIM,)):
 class SinParam:
   def __init__(self, name, shape, datatype="noise"):
     noise = LerpParam(name + '_noise', shape=shape, datatype=datatype)
-    orbit_radius = InterpolatorParam(name=name + '_radius', value=0.1)
+    orbit_radius = InterpolatorParam(name=name + '_radius', value=0.25)
     orbit_speed = InterpolatorParam(name=name + '_speed', value=1.0)
     orbit_time = InterpolatorParam(name=name + '_time', value=0.0)
     output = tf.math.sin(orbit_time.variable + noise.output) * orbit_radius.variable
@@ -110,7 +110,7 @@ class LerpParam:
       a = InterpolatorParam(name=name + '_a', shape=shape, datatype=datatype)
       b = InterpolatorParam(name=name + '_b', shape=shape, datatype=datatype)
     n = InterpolatorParam(name=name + '_n', value=0.0)
-    speed = InterpolatorParam(name=name + '_speed', value=0.1)
+    speed = InterpolatorParam(name=name + '_speed', value=1.0)
     output = a.variable * (1 - n.variable) + b.variable * n.variable
     interpolator.lerp_params[name] = self
     self.name = name
@@ -287,7 +287,11 @@ class Listener:
     self.rpc_client.send_status('processing', True)
     dt = 1 / FPS
     for i in range(99999):
-      if (i % 100) == 0 or i == 1:
+      if i == 0:
+        print("Loading network")
+      elif i == 1:
+        print("Processing")
+      elif (i % 100) == 0:
         print("Step {}".format(i))
       gen_time = time.time()
       gen_images = interpolator.on_step(i, dt, self.sess)