deepglint / unicom

[ICLR 2023] Unicom: Universal and Compact Representation Learning for Image Retrieval

Home Page:https://arxiv.org/pdf/2304.05884.pdf

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

是否会公开完整训练代码?

KeyaoZhao opened this issue · comments

您好,想请问下后续是否有打算公开预训练模型的训练代码呢?非常感谢!

现有的代码 retrirval.py 已经可以支持,两个维度的采样和1000w类别稀疏分类,分布式训练,可以使用retrieval.py进行预训练。

后续会更新数据集制作和pretrain 详细的README。

好的非常感谢!期待后续更详细的readme~

@anxiangsir 请问pretrain详细的README在哪呢?