使用有监督对比学习对CIFAR10进行分类
说明在https://zhuanlan.zhihu.com/p/665802511
1.run data.py下载数据集到cifar/
2.run main.py --method="xxx"开始训练
xxx可以为sl,scl,hybird,其他参数在args.py中调节
zhenglianchi / contrastive_learning_cifar10 Goto Github PK
View Code? Open in Web Editor NEW使用有监督对比学习对CIFAR10进行分类