| 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 |