print patch id while testing

main_sphere_compression.py

@@ -254,6 +254,7 @@ def test_epoch(test_dataloader, struct_loader, model, criterion, patch_res=None,
 
                     diff = (out_net["x_hat"] - decompressed["x_hat"]).abs()
                     diff_in_bits = (theoretical_rates - actual_rates).abs()
+                    print(f"Patch [{patch_id:03d}/{n_patches:03d}]")
                     print(f"max difference={diff.max()}, min difference={diff.min()}")
                     print(f"diff in bits={diff_in_bits}, ratio (compressed/training)={torch.div(actual_rates, theoretical_rates)}%", flush=True)