| ### DEPRECATED - use hydra conf instead ###### | |
| import torch | |
| import os | |
| # --------------------------------------- # | |
| VERSION = "1.24" | |
| # --------------------------------------- # | |
| ENCODER = "MERT" | |
| # - - - | |
| # MERT | |
| # M2L | |
| # LIBROSA | |
| # - - - | |
| # Encodec | |
| # DAC | |
| # --------------------------------------- # | |
| SEGMENT = "all" | |
| # all | |
| # f10s - first 10s | |
| # f30s - first 30s | |
| # 10s | |
| # 30s | |
| AGGREGATION_METHOD = "mean" | |
| # mean | |
| # median | |
| # 80th_percentile | |
| # max | |
| # --------------------------------------- # | |
| CLASSIFIER = "linear-mt" | |
| # transformer | |
| # linear | |
| # linear-small | |
| # linear-multitask | |
| # linear-small-multitask | |
| # linear-mt (mert-like classifier) | |
| # | |
| # --------------------------------------- # | |
| CHECKPOINT = "tb_logs/train_audio_classification/version_110/checkpoints/21-0.1202.ckpt" | |
| # --------------------------------------- # | |
| BATCH_SIZE = 8 | |
| N_EPOCHS = 50 | |
| # --------------------------------------- # | |
| GENRE_CLASS_SIZE = 87 | |
| MOOD_CLASS_SIZE = 56 | |
| INSTR_CLASS_SIZE = 40 | |
| DAC_LATENTS_SIZE = 72 | |
| DAC_RVQ_SIZE = 9 | |
| # --------------------------------------- # | |