Skip to content
Snippets Groups Projects
Commit 060bc6cf authored by Peter Pao-Huang's avatar Peter Pao-Huang
Browse files

Changed dataloader naming in evaluate to match distiller specs

parent 9870739c
No related branches found
No related tags found
No related merge requests found
......@@ -142,7 +142,7 @@ def get_loss(model: nn.Module, dataloader: DataLoader):
@torch.no_grad()
def evaluate(model: nn.Module, dataloader: DataLoader = None):
def evaluate(model: nn.Module, test_dataloader: DataLoader = None):
model.eval()
correct = 0
total = 0
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment