change model

script.py

@@ -53,11 +53,11 @@ print('Define Model')
 # model.load_state_dict(torch.load(model_path, map_location=device))
 
 # RESNET MODEL
-model = ResNet_LogSpec(sample_rate=24000, return_emb=False).to(device)
-model_path = './checkpoints/RESNET_LOGSPEC_ALL_DATA_FS_24000.pth'
+# model = ResNet_LogSpec(sample_rate=24000, return_emb=False).to(device)
+# model_path = './checkpoints/RESNET_LOGSPEC_ALL_DATA_FS_24000.pth'
 
-# model = ResNet_MelSpec(sample_rate=24000, return_emb=False).to(device)
-# model_path = './checkpoints/RESNET_MELSPEC_ALL_DATA_FS_24000.pth'
+model = ResNet_MelSpec(sample_rate=24000, return_emb=False).to(device)
+model_path = './checkpoints/RESNET_MELSPEC_ALL_DATA_FS_24000.pth'
 
 ## LCNN MODEL
 # model = LCNN(return_emb=False, fs=24000).to(device)

src/resnet_model.py

@@ -102,7 +102,7 @@ class ResNet_MelSpec(nn.Module):
     def __init__(self, sample_rate=16000, return_emb=False, num_class=2):
         super(ResNet_MelSpec, self).__init__()
 
-        self.threshold = 0.6
+        self.threshold = 0.4
         self.return_emb = return_emb
 
         if sample_rate == 16000: