From 6f03357342fb14b310642c57bc168905f3be2e1f Mon Sep 17 00:00:00 2001
From: FelixChan <2223485532@qq,com>
Date: Thu, 25 Sep 2025 15:48:48 +0800
Subject: [PATCH] 0925 add wandb proxy

---
 Amadeus/trainer_accelerate.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Amadeus/trainer_accelerate.py b/Amadeus/trainer_accelerate.py
index 3b7ccc2..3af0b6b 100644
--- a/Amadeus/trainer_accelerate.py
+++ b/Amadeus/trainer_accelerate.py
@@ -34,6 +34,8 @@ from .symbolic_encoding.decoding_utils import MidiDecoder4REMI
 from .evaluation_utils import add_conti_in_valid
 from .train_utils import NLLLoss4REMI
 
+os.environ['WANDB_INIT_TIMEOUT'] = '600'
+os.environ["WANDB_BASE_URL"] = "https://api.bandw.top"
 from data_representation.vocab_utils import LangTokenVocab
 class InfiniteSampler(Sampler):
     def __init__(self, data_source, shuffle=True):