From 5595366c9dfeefebba7c96729bcc29ff01f82f44 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 10 Jan 2020 20:00:21 +0100 Subject: test label training as part of encoding process --- cli/app/search/live.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'cli') 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(): -- cgit v1.2.3-70-g09d2