swuxyj / DeepHash-pytorch

Implementation of Some Deep Hash Algorithms, Including DPSH、DSH、DHN、HashNet、DSDH、DTSH、DFH、GreedyHash、CSQ.
MIT License
495 stars 116 forks source link

Mini batch HashNet. #36

Closed fuchun-wang closed 2 years ago

fuchun-wang commented 2 years ago

更改了 HashNet 计算 loss 的方式. 更改之前,计算 similarity 需要和全体训练数据进行计算。更改之后只需要和当前 batch 内的样本进行计算。 加速了训练过程,使得 HashNet 在大数据集上训练速度提升。 测试了 cifar10-1 和 imagenet100 两个数据集,精度无变化。