
SimCLR项目提供了一个PyTorch实现,展示了视觉表示形式对比学习的简洁框架及其源代码。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
PyTorch SimCLR 提供了一个简洁明了的视觉表示对比学习框架,并附带了详尽的文档资料。请参考提供的链接进行进一步了解。为了顺利安装,执行以下命令:$ conda env create --name simclr --file env.yml$,随后通过 $ conda activate simclr$ 激活该环境。在启动 SimCLR 之前,务必确认您已选择合适的运行配置,可以通过传递关键字参数到 run.py 文件来灵活地调整正在使用的配置。 具体而言,使用 $ python run.py - data / datasets - - dataset name stl10 - - log every n - steps 100 - - epochs 100$ 命令执行相应的设置。 如果您希望在 CPU 上进行实验(通常用于调试目的),请添加 --disable-cuda 选项。 为了实现 16 位精度 GPU 训练,请确保已安装并启用 --fp16_precision 标志。评估功能采用线性模型协议对学习到的特征进行分析。 首先,我们利用 SimCLR 在 STL10 无监督数据集上学习了这些特征表示。随后,我们在 SimCLR 的冻结特征向量之上训练了一个线性分类器。 通过对 STL10 数据集进行的这一训练过程...
全部评论 (0)
还没有任何评论哟~


