diff options
| -rw-r--r-- | cli/app/search/live.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py index 24d49e6..a1f5fd3 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -72,7 +72,6 @@ def label_sampler(num_classes=1, shape=(BATCH_SIZE, N_CLASS,)): for _ in range(int(num_classes)): j = random.randint(0, shape[1]-1) label[i, j] = random.random() - print("class: {} {}".format(j, label[i, j])) label[i] /= label[i].sum() return label @@ -299,9 +298,7 @@ class Interpolator: new_latent = np.expand_dims(data['latent'], axis=0) latent_stored = self.lerp_params['latent_stored'] - latent_stored_mix = self.lerp_params['latent_stored_mix'] label_stored = self.lerp_params['label_stored'] - label_stored_mix = self.lerp_params['label_stored_mix'] encoding_stored = self.lerp_params['encoding_stored'] encoding_stored_mix = self.lerp_params['encoding_stored_mix'] @@ -323,7 +320,7 @@ class Interpolator: encoding_stored.b.assign(new_encoding) latent_stored.b.assign(new_latent) label_stored.b.assign(new_label) - encoding_mix.switch() + encoding_stored_mix.switch() def on_step(self, i, dt, sess): for param in self.sin_params.values(): |