Update app.py

app.py

@@ -259,35 +259,16 @@ def get_script_args_info(exclude_args=None):
         {
             "name": "--learning_rate_warmup_fraction",
             "type": "slider",
-            "default": 0.25,
+            "default": 0.01,
             "min": 0.0,
             "max": 1.0,
             "step": 0.01,
             "help": "Fraction of iterations that the learning rate is increasing (warmup)",
         },
-        # {
-        #     "name": "--init_tau",
-        #     "type": "number",
-        #     "default": 1.0,
-        #     "help": "Initial tau value for Gumbel-Softmax",
-        # },
-        # {
-        #     "name": "--final_tau",
-        #     "type": "number",
-        #     "default": 0.01,
-        #     "help": "Final tau value for Gumbel-Softmax",
-        # },
-        # {
-        #     "name": "--min_layers",
-        #     "type": "number",
-        #     "default": 0,
-        #     "precision": 0,
-        #     "help": "Minimum number of layers. Used for pruning.",
-        # },
         {
             "name": "--early_stopping",
             "type": "number",
-            "default": 1500,
+            "default": 5000,
             "precision": 0,
             "help": "Number of steps without improvement before stopping",
         },
@@ -314,6 +295,13 @@ def get_script_args_info(exclude_args=None):
             "precision": 0,
             "help": "Number of rounds to choose the starting height map from.",
         },
+        {
+            "name": "--num_init_cluster_layers",
+            "type": "number",
+            "default": 35,
+            "precision": 0,
+            "help": "Number of layer clusters for the initialization.",
+        },
     ]
     return [arg for arg in all_args_info if arg["name"] not in exclude_args]