search

zhvng / open-musiclm

Implementation of MusicLM, a text to music model published by Google Research, with a few modifications.
https://arxiv.org/abs/2301.11325
MIT License
511 stars 59 forks source link

ClapRVQTrainer does not have log_with #10

Closed heury closed 1 year ago

heury commented 1 year ago

ClapRVQTrainer does not have "log_with" and it causes error in following line

if 'tensorboard' in self.log_with: self.accelerator.init_trackers(f"claprvq{int(time.time() * 1000)}", config=hps) else: self.accelerator.init_trackers(f"clap_rvq", config=hps)

So, I added log_with refer to SingleStageTrainer class and it works fine.

self.log_with = accelerate_kwargs['log_with'] if 'log_with' in accelerate_kwargs else None

zhvng commented 1 year ago

my bad! fixed in 0e3ada6