AhmadMoussa

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider, Button, RadioButtons
import librosa

fig, ax = plt.subplots()
plt.subplots_adjust(left=0.1, bottom=0.35)

samples = 1
audio, sr = librosa.core.load("Snare.wav") # load your audio here

AhmadMoussa

import torch
from torch import nn

def convBlock(inc, outc, ksz, conv_or_deconv):
    return nn.Sequential(
        nn.Conv2d(in_channels=inc, out_channels=outc, kernel_size=ksz,
                  stride=2) if conv_or_deconv else nn.ConvTranspose2d(in_channels=inc, out_channels=outc,
                                                                      kernel_size=ksz, stride=2),
        nn.LeakyReLU(),
        nn.BatchNorm2d(num_features=outc)