Minor Fixes

Author: federicobrancasi

DeepQuant/Pipeline/DequantUnify.py

@@ -69,9 +69,10 @@ def mergeDequants(
     unifiedModel = unifyLinearDequants(model, debug=debug)
     unifiedModel.recompile()
 
-    print(cc.header("4. Network after Modification of Dequant Nodes"))
-    printer.printTabular(unifiedModel)
-    print()
+    if debug:
+        print(cc.header("4. Network after Modification of Dequant Nodes"))
+        printer.printTabular(unifiedModel)
+        print()
 
     with torch.no_grad():
         output = unifiedModel(exampleInput)

Tests/TestResNet18.py

@@ -222,9 +222,9 @@ def deepQuantTestResnet18() -> None:
     dataset.targets = [s[1] for s in newSamples]
 
     # FBRANCASI: Optional, reduce number of example for faster validation
-    # DATASET_LIMIT = 256
-    # dataset = Subset(dataset, list(range(DATASET_LIMIT)))
-    # print(f"Validation dataset size set to {len(dataset)} images.")
+    DATASET_LIMIT = 256
+    dataset = Subset(dataset, list(range(DATASET_LIMIT)))
+    print(f"Validation dataset size set to {len(dataset)} images.")
 
     calibLoader = DataLoader(
         Subset(dataset, list(range(256))), batch_size=32, shuffle=False, pin_memory=True