VoyagerYuan commited on
Commit
c72b8ff
·
1 Parent(s): 2f88585

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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