VoyagerYuan
commited on
Commit
·
c72b8ff
1
Parent(s):
2f88585
Update app.py
Browse files
app.py
CHANGED
@@ -204,8 +204,8 @@ class MultiMultiSignalingGame:
|
|
204 |
|
205 |
|
206 |
def train_signal_game(NUM_SENDERS, NUM_RECEIVERS, num_rounds):
|
207 |
-
para_checker = st.empty()
|
208 |
-
para_checker.text(f"NUM_SENDERS: {NUM_SENDERS}, NUM_RECEIVERS: {NUM_RECEIVERS}, num_rounds: {num_rounds}, EMBEDDING_DIM: {EMBEDDING_DIM}, HIDDEN_DIM: {HIDDEN_DIM}, LATENT_DIM: {LATENT_DIM}, SEQ_LEN: {SEQ_LEN}, TAU: {TAU}, nhead: {NHEAD}, num_layers: {NUM_LAYERS}, BATCH_SIZE: {BATCH_SIZE}")
|
209 |
senders = [TransformerEncoder().to(device) for _ in range(NUM_SENDERS)]
|
210 |
receivers = [TransformerDecoder().to(device) for _ in range(NUM_RECEIVERS)]
|
211 |
|
|
|
204 |
|
205 |
|
206 |
def train_signal_game(NUM_SENDERS, NUM_RECEIVERS, num_rounds):
|
207 |
+
# para_checker = st.empty()
|
208 |
+
# para_checker.text(f"NUM_SENDERS: {NUM_SENDERS}, NUM_RECEIVERS: {NUM_RECEIVERS}, num_rounds: {num_rounds}, EMBEDDING_DIM: {EMBEDDING_DIM}, HIDDEN_DIM: {HIDDEN_DIM}, LATENT_DIM: {LATENT_DIM}, SEQ_LEN: {SEQ_LEN}, TAU: {TAU}, nhead: {NHEAD}, num_layers: {NUM_LAYERS}, BATCH_SIZE: {BATCH_SIZE}")
|
209 |
senders = [TransformerEncoder().to(device) for _ in range(NUM_SENDERS)]
|
210 |
receivers = [TransformerDecoder().to(device) for _ in range(NUM_RECEIVERS)]
|
211 |
|