File size: 637 Bytes
d967830 |
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 |
class Config:
model_name = "BirdAST_Baseline_5folds"
backbone_name = "MIT/ast-finetuned-audioset-10-10-0.4593"
n_classes = 728 # number of classes in the dataset
audio_sr = 16000 #Hz
segment_length = 10 #s
fft_window = 0.025 #s
hop_window_length = 0.01 #s
n_mels = 128
low_cut = 1000 #Hz
high_cut = 8000 #Hz
top_db = 100
batch_size = 4
num_workers = 0
n_splits = 5
max_lr = 1e-5
epochs = 15
weight_decay = 0.01
lr_final_div = 1000
amp = True
grad_accum_steps = 1
max_grad_norm = 1e7
print_epoch_freq = 1
print_freq = 500
random_seed = 2046 |