From 060bc6cf74c089f46ae6687aaf507564a2520008 Mon Sep 17 00:00:00 2001
From: Peter Pao-Huang <ytp2@miranda.cs.illinois.edu>
Date: Mon, 12 Jul 2021 23:28:37 -0500
Subject: [PATCH] Changed dataloader naming in evaluate to match distiller
 specs

---
 hpvm/test/epoch_dnn/torch_dnn/quantizer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
index ed94754d6a..729e8bd356 100644
--- a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
+++ b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
@@ -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
-- 
GitLab