SimCLR:SimCLR的PyTorch实现:视觉表示形式对比学习的简单框架 源码

PyTorch SimCLR:视觉表示形式对比学习的简单框架 带有完整文档的博客文章: 另请参见。 安装 $ conda env create --name simclr --file env.yml $ conda activate simclr $ python run.py 配置文件 在运行SimCLR之前,请确保选择正确的运行配置。 您可以通过将关键字参数传递给run.py文件来更改正在运行的配置。 $ python run . py - data . / datasets - - dataset - name stl10 - - log - every - n - steps 10