SimCLR:SimCLR的PyTorch实现

时间:2024-04-09 09:43:08
【文件属性】:

文件名称:SimCLR:SimCLR的PyTorch实现

文件大小:248KB

文件格式:ZIP

更新时间:2024-04-09 09:43:08

pytorch representation-learning unsupervised-learning simclr contrastive-learning

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 ||


【文件预览】:
SimCLR-master
----setup.py(4KB)
----.gitignore(84B)
----media()
--------architecture.png(231KB)
--------lr_cosine_decay_schedule.png(20KB)
--------tensorboard.png(12KB)
----requirements.txt(24B)
----linear_evaluation.py(6KB)
----LICENSE(1KB)
----utils()
--------__init__.py(47B)
--------yaml_config_hook.py(709B)
----README.md(7KB)
----run_all.sh(187B)
----config()
--------config.yaml(1006B)
----setup_apex.sh(379B)
----environment.yml(87B)
----setup.sh(95B)
----main.py(5KB)
----simclr()
--------__init__.py(27B)
--------modules()
--------simclr.py(1KB)
----main_pl.py(4KB)
----model.py(1KB)

网友评论