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

add ability to skip prompt asking whether to clear previous results, by... 

add ability to skip prompt asking whether to clear previous results, by setting force_clear_prev_results to something other than None
parent df17dcd0

musiclm_pytorch/trainer.py

+3 −2
Original line number Diff line number Diff line
@@ -160,7 +160,8 @@ class MuLaNTrainer(nn.Module): 
        random_split_seed = 42,

        save_model_every = 1000,

        results_folder = './results',

        accelerate_kwargs: dict = dict()

        accelerate_kwargs: dict = dict(),

        force_clear_prev_results = None  # set to True | False to skip the prompt

    ):

        super().__init__()

        assert batch_size > 1, 'batch size must be greater than 1 for contrastive learning (but ideally as large as possible)'
@@ -240,7 +241,7 @@ class MuLaNTrainer(nn.Module): 


        self.results_folder = Path(results_folder)



        if len([*self.results_folder.glob('**/*')]) > 0 and yes_or_no('do you want to clear previous experiment checkpoints and results?'):

        if force_clear_prev_results is True or (not exists(force_clear_prev_results) and len([*self.results_folder.glob('**/*')]) > 0 and yes_or_no('do you want to clear previous experiment checkpoints and results?')):

            rmtree(str(self.results_folder))



        self.results_folder.mkdir(parents = True, exist_ok = True)

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

  version = '0.0.19',

  license='MIT',

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

  author = 'Phil Wang',