| batch_size: 48 | |
| mel_frames: 800 | |
| lr: 0.0001 | |
| num_epochs: 100000 | |
| sample_interval: 250 | |
| h5_file_path: './dataset/gtzan_train.h5' | |
| device: 'cuda:4' | |
| input_size: [80, 800] | |
| patch_size: 8 | |
| in_channels: 1 | |
| hidden_size: 384 | |
| depth: 12 | |
| num_heads: 6 | |
| checkpoint_dir: 'gtzan-ck' | |