Commit f2dc4628 authored by Phil Wang's avatar Phil Wang
Browse files

in contrastive learning, make sure always the full batch

parent 0d1f2b3d

musiclm_pytorch/trainer.py

+2 −2
Original line number Diff line number Diff line
@@ -203,9 +203,9 @@ class MuLaNTrainer(nn.Module): 


        # dataloader



        self.dl = get_dataloader(self.ds, batch_size = batch_size, shuffle = True, pad_to_longest = False)

        self.dl = get_dataloader(self.ds, batch_size = batch_size, shuffle = True, pad_to_longest = False, drop_last = True)



        self.valid_dl = get_dataloader(self.valid_ds, batch_size = batch_size, shuffle = True, pad_to_longest = False)

        self.valid_dl = get_dataloader(self.valid_ds, batch_size = batch_size, shuffle = True, pad_to_longest = False, drop_last = True)



        # prepare with accelerator

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.16',

  version = '0.0.17',

  license='MIT',

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

  author = 'Phil Wang',