✨ feat(monitor): Add support for tensorboard

KarhouTam · May 18, 2024 · 05aca78 · 05aca78
1 parent 1c5cbe3
commit 05aca78
Show file tree

Hide file tree

Showing 5 changed files with 340 additions and 113 deletions.
diff --git a/.environment/poetry.lock b/.environment/poetry.lock
diff --git a/.environment/pyproject.toml b/.environment/pyproject.toml
@@ -4,13 +4,13 @@ version = "0.2.0"
 description = "Benchmark of federated learning that aim solving image classification tasks."
 authors = ["KarhouTam <karhoutam@qq.com>"]
 license = "GPL-2.0-or-later"
-readme = ["README.md", "data/README.md"]
+readme = ["../README.md", "../data/README.md"]
 repository = "https://github.com/KarhouTam/FL-bench"
 
 [[tool.poetry.source]]
 name = "mirrors"
 url = "https://mirrors.sustech.edu.cn/pypi/simple"
-priority = "default"
+priority = "primary"
 
 [tool.poetry.dependencies]
 python = ">=3.10, <3.12"
@@ -29,6 +29,7 @@ faiss-cpu = "1.7.4"
 pynvml = "11.5.0"
 PyYAML = "6.0.1"
 ray = "2.10.0"
+tensorboard = "2.16.2"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/.environment/requirements.txt b/.environment/requirements.txt
@@ -12,4 +12,5 @@ faiss-cpu~=1.7.4
 scikit-learn~=1.4.0
 pynvml~=11.5.0
 PyYAML~=6.0.1
-ray[default]~=2.10.0
+ray[default]~=2.10.0
+tensorboard~=2.16.2