Update app.py

app.py

@@ -273,28 +273,28 @@ NUM_SENDERS = st.sidebar.slider("NUM_SENDERS", 1, 10, 2)
 NUM_RECEIVERS = st.sidebar.slider("NUM_RECEIVERS", 1, 10, 2)
 num_rounds = st.sidebar.slider("num_rounds", 1000, 100000, 10000, 1000)
 
-use_cosine_annealing = st.sidebar.checkbox("Use Annealing")
-if use_cosine_annealing:
-    annealing_strategy = st.sidebar.selectbox("Annealing Strategy", ["linear", "cosine"])
-    TAU = st.sidebar.slider("Start Temp.", 0.1, 10.0, 1.0)
-    final_tau = st.sidebar.slider("Final Temp.", 0.1, 10.0, 1.0)
-else:
-    annealing_strategy = None
-    TAU = st.sidebar.slider("TAU", 0.1, 10.0, 1.0)
-
-optimizer_options = ["Adam", "AdamW", "SGD"]
-OPTMIZER = st.sidebar.selectbox("Optimizer", optimizer_options)
-LEARNING_RATE = st.sidebar.slider("Learning Rate", 1e-5, 1e-2, 1e-3)
-
-
-EMBEDDING_DIM = st.sidebar.slider("EMBEDDING_DIM", 1, 128, 16)
-HIDDEN_DIM = st.sidebar.slider("HIDDEN_DIM", 1, 128, 16)
-LATENT_DIM = st.sidebar.slider("LATENT_DIM", 1, 128, 16)
-SEQ_LEN = st.sidebar.slider("SEQ_LEN", 1, 128, 16)
-# TAU = st.sidebar.slider("TAU", 0.1, 10.0, 1.0)
-NHEAD = st.sidebar.slider("nhead", 1, 8, 4)
-NUM_LAYERS = st.sidebar.slider("num_layers", 1, 6, 2)
-BATCH_SIZE = st.sidebar.slider("BATCH_SIZE", 1, 128, 32)
+advanced_settings = st.sidebar.expander("Advanced settings")
+with advanced_settings:
+    use_cosine_annealing = st.checkbox("Use Annealing")
+    if use_cosine_annealing:
+        annealing_strategy = st.selectbox("Annealing Strategy", ["linear", "cosine"])
+        TAU = st.slider("Start Temp.", 0.1, 10.0, 1.0)
+        final_tau = st.slider("Final Temp.", 0.1, 10.0, 1.0)
+    else:
+        annealing_strategy = None
+        TAU = st.slider("TAU", 0.1, 10.0, 1.0)
+
+    optimizer_options = ["Adam", "AdamW", "SGD"]
+    OPTMIZER = st.selectbox("Optimizer", optimizer_options)
+    LEARNING_RATE = st.slider("Learning Rate", 1e-5, 1e-2, 1e-3)
+
+    EMBEDDING_DIM = st.slider("EMBEDDING_DIM", 1, 128, 16)
+    HIDDEN_DIM = st.slider("HIDDEN_DIM", 1, 128, 16)
+    LATENT_DIM = st.slider("LATENT_DIM", 1, 128, 16)
+    SEQ_LEN = st.slider("SEQ_LEN", 1, 128, 16)
+    NHEAD = st.slider("nhead", 1, 8, 4)
+    NUM_LAYERS = st.slider("num_layers", 1, 6, 2)
+    BATCH_SIZE = st.slider("BATCH_SIZE", 1, 128, 32)
 
 if st.sidebar.button('Start'):
     train_signal_game(NUM_SENDERS, NUM_RECEIVERS, num_rounds)