From 9870739cef9e9c6fd815158646a941bb3d16f644 Mon Sep 17 00:00:00 2001
From: Peter Pao-Huang <ytp2@miranda.cs.illinois.edu>
Date: Mon, 12 Jul 2021 23:24:14 -0500
Subject: [PATCH] Changed evaluate method to conform to distiller requirements

---
 hpvm/test/epoch_dnn/torch_dnn/quantizer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
index a473c0a057..ed94754d6a 100644
--- a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
+++ b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
@@ -54,7 +54,7 @@ def quantize(
     if not os.path.isfile(stats_file):
         # generates `stats_file`
         collect_quant_stats(
-            model, lambda model: get_loss(model, dataloader), save_dir=working_dir
+            model, get_loss, dataloader, save_dir=working_dir
         )
 
     # Generate Quantized Scales
@@ -142,7 +142,7 @@ def get_loss(model: nn.Module, dataloader: DataLoader):
 
 
 @torch.no_grad()
-def evaluate(model: nn.Module, dataloader: DataLoader):
+def evaluate(model: nn.Module, dataloader: DataLoader = None):
     model.eval()
     correct = 0
     total = 0
-- 
GitLab