求!论文BYOL训练cifar10数据集1000个epoch的模型

问题遇到的现象和发生背景

读取BYOL原论文之后,找了一篇认可度较高的用pytorch实现的代码,自己训练1000次,利用得到的图像特征去做k-means聚类,聚类准确度很低,只有27%左右。cifar10数据集一般可以做到80%以上,这个结果有点低了。问一下大家有没有用这个代码训练一个准确度较高的模型,不同代码也可以,用于深度图像聚类。

代码:https://github.com/lucidrains/byol-pytorch

img

问题相关代码

训练代码

img

模型评估

img

img