File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -23,7 +23,7 @@ modules.clap.CLAP.text_encoder_name = "/gpfs/scratch/upf97/model_weights/all-mpn
2323modules.clap.CLAP.audio_encoder_params = {
2424 "encodec_weights_path": "/gpfs/scratch/upf97/model_weights/encodec_24khz/"
2525 }
26- modules.clap.CLAP.proj_size = 512
26+ modules.clap.CLAP.proj_size = 768
2727modules.clap.CLAP.temp = 0.1
2828modules.clap.CLAP.lr = 1e-4
2929modules.clap.CLAP.weight_decay = 1e-2
@@ -40,7 +40,7 @@ CosineAnnealingCallback.eta_min = 1e-7
4040build_dev_datamodule.datamodule = @discotube_text_audio
4141
4242# Discogs datamodule parameters
43- DiscotubeTextAudioDataModule.batch_size = 128
43+ DiscotubeTextAudioDataModule.batch_size = 64
4444DiscotubeTextAudioDataModule.num_workers = 20
4545DiscotubeTextAudioDataModule.data_dir = "/"
4646DiscotubeTextAudioDataModule.filelist_train = "/gpfs/projects/upf97/data/train_mmap.txt"
You can’t perform that action at this time.
0 commit comments