From 2e8dc0933f31bf449ecc11ac1b4dc1833fdaad42 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 20 二月 2024 18:01:15 +0800
Subject: [PATCH] train finetune

---
 setup.py |    5 ++---
 1 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index f7e6ee6..f703bb4 100644
--- a/setup.py
+++ b/setup.py
@@ -37,15 +37,14 @@
         # "textgrid",
         # "protobuf",
         "tqdm",
-        "hdbscan",
-        "umap",
+        "umap_learn",
         "jaconv",
         "hydra-core>=1.3.2",
     ],
     # train: The modules invoked when training only.
     "train": [
         "editdistance",
-        "wandb",
+        "tensorboardX",
     ],
     # all: The modules should be optionally installled due to some reason.
     #      Please consider moving them to "install" occasionally

--
Gitblit v1.9.1