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):