Spaces:

alisrbdni
/

federated-learning-dynamic

Sleeping

alisrbdni commited on May 18, 2024

Commit

bde5538

verified ·

1 Parent(s): f001673

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -234,7 +234,7 @@ import streamlit
 # If you're curious of all the loggers
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 fl.common.logger.configure(identifier="myFlowerExperiment", filename="./log.txt")
 def load_data(dataset_name, train_size=20, test_size=20, num_clients=2):

 # If you're curious of all the loggers
+DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 fl.common.logger.configure(identifier="myFlowerExperiment", filename="./log.txt")
 def load_data(dataset_name, train_size=20, test_size=20, num_clients=2):