search

Zhongdao / gcn_clustering

Code for CVPR'19 paper Linkage-based Face Clustering via GCN
MIT License
360 stars 86 forks source link

Update test.py #38

Closed espectre closed 3 years ago

espectre commented 3 years ago

while the last batch is single ,line 154 will cause bug for 'squeeze()',so we need to delete it.