File size: 258 Bytes
070e26e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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'
|