lingyu123-su/MIDIFoundationModel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

0925 add wandb proxy

Browse Source
This commit is contained in:
FelixChan
2025-09-25 15:48:48 +08:00
parent 5b8bfd70e3
commit 6f03357342
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Amadeus/trainer_accelerate.py
View File

@ -34,6 +34,8 @@ from .symbolic_encoding.decoding_utils import MidiDecoder4REMI
from .evaluation_utils import add_conti_in_valid from .evaluation_utils import add_conti_in_valid
from .train_utils import NLLLoss4REMI 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 from data_representation.vocab_utils import LangTokenVocab
class InfiniteSampler(Sampler): class InfiniteSampler(Sampler):
def __init__(self, data_source, shuffle=True): def __init__(self, data_source, shuffle=True):