def __init__(self, classes, config={}, state_dict=None):
    super(AudioCRNN, self).__init__(config)
    in_chan = (2 if (config['transforms']['args']['channels'] == 'stereo') else 1)
    self.classes = classes
    self.lstm_units = 64
    self.lstm_layers = 2
    self.spec = MelspectrogramStretch(hop_length=64, num_mels=80, fft_length=1024, norm='whiten', stretch_param=[0.4, 0.4])
    self.net = parse_cfg(config['cfg'], in_shape=[in_chan, self.spec.n_mels, 400])
def __init__(self, classes, config={}, state_dict=None):
    super(AudioCRNN, self).__init__(config)
    in_chan = (2 if (config['transforms']['args']['channels'] == 'stereo') else 1)
    self.classes = classes
    self.lstm_units = 64
    self.lstm_layers = 2
    self.spec = MelspectrogramStretch(hop_length=64, num_mels=80, fft_length=1024, norm='whiten', stretch_param=[0.4, 0.4])
    self.net = parse_cfg(config['cfg'], in_shape=[in_chan, self.spec.n_mels, 400])