Spaces:
Sleeping
Sleeping
File size: 1,034 Bytes
6ad6801 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
### 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
# --------------------------------------- #
|