update test

SuchitReddi · Jan 13, 2021 · 8892187 · 8892187
1 parent e9a8c77
commit 8892187
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test.py b/test.py
@@ -24,8 +24,8 @@ def test_model(feat_model_path, loss_model_path, part, add_loss, device):
  loss_model = torch.load(loss_model_path) if add_loss != "softmax" else None
  # if add_loss != "softmax":
  # loss_model = nn.DataParallel(loss_model, list(range(torch.cuda.device_count())))
- test_set = ASVspoof2019("LA", "/data/neil/DS_10283_3336/", "/dataNVME/neil/ASVspoof2019LAFeatures/",
- "/data/neil/DS_10283_3336/LA/ASVspoof2019_LA_cm_protocols/", part,
+ test_set = ASVspoof2019("LA", "/dataNVME/neil/ASVspoof2019LAFeatures/",
+ "/data/neil/DS_10283_3336/PA/ASVspoof2019_LA_cm_protocols/", part,
  "LFCC", feat_len=750, padding="repeat")
  testDataLoader = DataLoader(test_set, batch_size=32, shuffle=False, num_workers=0,
  collate_fn=test_set.collate_fn)