Added wandb tracking to SemanticTransformerTrainer, CoarseTransformerTrainer, and FineTransformerTrainer

lucidrains / audiolm-pytorch

Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Research, in Pytorch

MIT License

2.32k stars 249 forks source link

Added wandb tracking to SemanticTransformerTrainer, CoarseTransformerTrainer, and FineTransformerTrainer #257

Closed LukasNel closed 6 months ago

LukasNel commented 6 months ago

Added optional wandb tracking to SemanticTransformerTrainer, CoarseTransformerTrainer, and FineTransformerTrainer to help record the runs better and make it more consistent with SoundstreamTransformerTrainer.

lucidrains commented 6 months ago

nice! great job!