Changed evaluate method to conform to distiller requirements

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