Commit bd916e97 authored by Verena Praher's avatar Verena Praher
Browse files

fix bug

parent 2b7b9351
......@@ -82,7 +82,7 @@ def ERF_generate(model, loader):
counter = 0
accum = None
for step, (data, _, targets) in enumerate(loader):
for step, (data, targets) in enumerate(loader):
data = data.cuda()
data.requires_grad = True
outputs = model(data)
......@@ -128,6 +128,9 @@ def run():
trainer.fit(model)
save(model, "last_model.pth")
dataset = MelSpecDataset(phase='test', ann_root=PATH_ANNOTATIONS, spec_root=PATH_MELSPEC_DOWNLOADED_FRAMED)
test_loader = DataLoader(dataset=dataset,
batch_size=32,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment