上传者: 42133415
|
上传时间: 2021-10-10 09:36:48
|
文件大小: 248KB
|
文件类型: -
SimCLR
SimCLR的PyTorch实现:T. Chen等人的可视表示形式对比学习的简单框架。包括对以下方面的支持:
分布式数据并行训练
全局批量归一化
LARS(分层自适应速率缩放)优化器。
在Google Colab Notebook中打开SimCLR(具有TPU支持)
在tensorboard.dev上打开SimCLR结果比较:
快速入门(微调线性分类器)
这将下载一个预训练的模型并训练线性分类器,该分类器在STL-10测试仪上应具有± 82.9%的精度。
git clone https://github.com/spijkervet/SimCLR.git && cd SimCLR
wget https://github.com/Spijkervet/SimCLR/releases/download/1.2/checkpoint_100.tar
sh setup.sh ||