From e036588c6d387d00c672b69ba4f0d43b4433c23c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kai=20Li=20=28=E6=9D=8E=E5=87=AF=29?= Date: Mon, 23 Sep 2024 00:17:21 +0800 Subject: [PATCH] Update train.py --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 9bb2a5f..e7ba4e6 100644 --- a/train.py +++ b/train.py @@ -23,7 +23,7 @@ from look2hear.utils import RankedLogger, instantiate, print_only import warnings warnings.filterwarnings("ignore") -os.environ['WANDB_API_KEY'] = "ca76d47c4da23aa9ceb8788307ab090ef5f5713c" + def train(cfg: DictConfig) -> Tuple[Dict[str, Any], Dict[str, Any]]: @@ -133,4 +133,4 @@ def train(cfg: DictConfig) -> Tuple[Dict[str, Any], Dict[str, Any]]: OmegaConf.save(cfg, os.path.join(cfg.exp.dir, cfg.exp.name, "config.yaml")) train(cfg) - \ No newline at end of file +