Update app.py

app.py

@@ -271,29 +271,29 @@ st.title('Multi-Agents Signaling Game')
 
 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)
+num_rounds = st.sidebar.slider("NUM_ROUNDS", 1000, 100000, 10000, 1000)
 
 advanced_settings = st.sidebar.expander("Advanced settings")
 with advanced_settings:
-    use_cosine_annealing = st.checkbox("Use Annealing")
+    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)
+        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)
+        TAU = st.slider("TEMP.", 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)
+    OPTMIZER = st.selectbox("OPTIMIZER", optimizer_options)
+    LEARNING_RATE = st.slider("LEARNING RATE", 1e-5, 1e-2, 1e-3, format="%.5f")
 
     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)
+    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'):