diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 20:00:21 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 20:00:21 +0100 |
| commit | 5595366c9dfeefebba7c96729bcc29ff01f82f44 (patch) | |
| tree | 62aff92e8dc196184f22cc8306d307fe64673221 | |
| parent | d92efc0d43e566f72fc5bdea9368fdaa0cc5f424 (diff) | |
test label training as part of encoding process
| -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(): |