config.json

{
  "model_type": "resnet18",
  "num_classes": 2,
  "class_names": [
    "clean",
    "knocking"
  ],
  "input_type": "mel_spectrogram",
  "input_size": [
    3,
    224,
    224
  ],
  "sample_rate": 16000,
  "n_mels": 128,
  "n_fft": 1024,
  "hop_length": 512
}