From 57e03c1bee2bcb60d25201243cdd51355714b893 Mon Sep 17 00:00:00 2001
From: Dushyant Rao <dushyantr@google.com>
Date: Thu, 27 Feb 2020 14:33:25 +0000
Subject: [PATCH] Fix default config parameters.

PiperOrigin-RevId: 297582671
---
 curl/train_unsup.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/curl/train_unsup.py b/curl/train_unsup.py
index 48e7ad22..ecb423ec 100644
--- a/curl/train_unsup.py
+++ b/curl/train_unsup.py
@@ -28,11 +28,11 @@
 def main(unused_argv):
   if FLAGS.dataset == 'mnist':
     n_y = 25
-    n_y_active = 25
+    n_y_active = 1
     n_z = 50
   else:  # omniglot
     n_y = 100
-    n_y_active = 100
+    n_y_active = 1
     n_z = 100
 
   training.run_training(
@@ -50,7 +50,7 @@ def main(unused_argv):
       train_supervised=False,
       n_steps=100000,
       report_interval=10000,
-      knn_values=[3, 5, 10],
+      knn_values=[3],
       random_seed=1,
       encoder_kwargs={
           'encoder_type': 'multi',
@@ -62,8 +62,8 @@ def main(unused_argv):
           'n_dec': [500],
           'dec_up_strides': None,
       },
-      dynamic_expansion=False,
-      ll_thresh=-0.0,
+      dynamic_expansion=True,
+      ll_thresh=-200.0,
       classify_with_samples=True,
       gen_replay_type=None,
       use_supervised_replay=False,