Commit 0d1f2b3d authored by Phil Wang's avatar Phil Wang
Browse files

batch size must be greater than 1 for contrastive learning

parent 6d45461a

musiclm_pytorch/trainer.py

+2 −0
Original line number Diff line number Diff line
@@ -163,6 +163,8 @@ class MuLaNTrainer(nn.Module): 
        accelerate_kwargs: dict = dict()

    ):

        super().__init__()

        assert batch_size > 1, 'batch size must be greater than 1 for contrastive learning (but ideally as large as possible)'



        self.accelerator = Accelerator(**accelerate_kwargs)



        self.mulan = mulan

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages 
setup(

  name = 'musiclm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.0.15',

  version = '0.0.16',

  license='MIT',

  description = 'MusicLM - AudioLM + Audio CLIP to text to music synthesis',

  author = 'Phil Wang',