mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-12 20:15:31 +08:00
refactor(compressors): consolidate pipeline and improve mask handling
This commit is contained in:
@@ -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)
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user