Update app.py
Browse files
app.py
CHANGED
@@ -229,6 +229,10 @@ from collections import OrderedDict
|
|
229 |
import flwr as fl
|
230 |
from logging import INFO, DEBUG
|
231 |
from flwr.common.logger import log
|
|
|
|
|
|
|
|
|
232 |
|
233 |
DEVICE = torch.device("cpu")
|
234 |
fl.common.logger.configure(identifier="myFlowerExperiment", filename="./log.txt")
|
@@ -350,6 +354,7 @@ def read_log_file():
|
|
350 |
return file.read()
|
351 |
|
352 |
def main():
|
|
|
353 |
st.write("## Federated Learning with Dynamic Models and Datasets for Mobile Devices")
|
354 |
dataset_name = st.selectbox("Dataset", ["imdb", "amazon_polarity", "ag_news"])
|
355 |
model_name = st.selectbox("Model", ["bert-base-uncased", "facebook/hubert-base-ls960", "distilbert-base-uncased"])
|
|
|
229 |
import flwr as fl
|
230 |
from logging import INFO, DEBUG
|
231 |
from flwr.common.logger import log
|
232 |
+
import logging
|
233 |
+
import streamlit
|
234 |
+
|
235 |
+
# If you're curious of all the loggers
|
236 |
|
237 |
DEVICE = torch.device("cpu")
|
238 |
fl.common.logger.configure(identifier="myFlowerExperiment", filename="./log.txt")
|
|
|
354 |
return file.read()
|
355 |
|
356 |
def main():
|
357 |
+
st.markdown(print(streamlit.logger._loggers))
|
358 |
st.write("## Federated Learning with Dynamic Models and Datasets for Mobile Devices")
|
359 |
dataset_name = st.selectbox("Dataset", ["imdb", "amazon_polarity", "ag_news"])
|
360 |
model_name = st.selectbox("Model", ["bert-base-uncased", "facebook/hubert-base-ls960", "distilbert-base-uncased"])
|