Each audio file was resampled to 16 kHz and downmixed to mono. The training batch size was 16, and for each audio file, we randomly selected a 2-second segment. On average, there were 112.5 segments in a track, and therefore training with 443,250 (=3,940 \(\times \)  112.5) items would be approximately one epoch. With a Nvidia Tesla P100 and a batch size of 32, it took about 9 hours to train a single epoch. We implemented DrummerNet using Pytorch 1.0 [1]} and used Librosa 0.6.3 [2]} and Madmom 0.16 [3]} for audio processing and peak-picking.
