Skip to content

Commit

Permalink
Update train.py
Browse files Browse the repository at this point in the history
  • Loading branch information
JusperLee authored Sep 22, 2024
1 parent e7ec6cd commit e036588
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]]:
Expand Down Expand Up @@ -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)


0 comments on commit e036588

Please sign in to comment.