refactor(compressors): consolidate pipeline and improve mask handling

This commit is contained in:
2026-03-26 19:00:13 +08:00
parent 90d5a8f08a
commit 968819e113
11 changed files with 302 additions and 121 deletions

View File

@@ -60,7 +60,7 @@ def _establish_eval_database(
{
"id": global_idx + j,
"label": labels_list[j],
"vector": all_features[global_idx + j].numpy(),
"vector": all_features[global_idx + j].detach().cpu().numpy(),
}
for j in range(batch_size)
]