remove score printing

Author: jiho264

quant/quantizer.py

@@ -257,7 +257,7 @@ def forward(self, x: torch.Tensor):
                     out = x_dq * 1 / i
 
                     score = lp_loss(x, out, p=2, reduction="all")
-                    print(f"scale: {i}, score: {score}")
+                    # print(f"scale: {i}, score: {score}")
 
                     if score < best_score:
                         best_score = score
@@ -271,6 +271,7 @@ def forward(self, x: torch.Tensor):
                 print(x_q.unique().numel(), x_q.unique())
                 print(x_dq.unique().numel(), x_dq.unique())
                 print(x.unique().numel(), x.unique())
+                print()
                 if int_max == 384:
                     assert x.unique().numel() <= 17
                 elif int_max == 256:

test_quant.py

@@ -54,7 +54,7 @@ def get_args_parser():
     )
     parser.add_argument(
         "--log_quant_scheme",
-        default="Sqrt2_17",
+        default="BitLog2_Half_17",
         choices=[
             "Sqrt2_16",
             "Sqrt2_17",  # Original RepQ-ViT