SimSiam:论文“探索简单的暹罗表示学习”的pytorch实现 源码

消息 已经两个月了,我想我终于找到了Simsiam / BYOL避免使用停止梯度和预测变量来避免崩溃解决方案的真正原因!!! 在上关注我,敬请期待! SimSiam Chenxinlei和He Kaiming的《一文的PyTorch实现 依存关系 如果您没有python 3环境: conda create -n simsiam python=3.8 conda activate simsiam 然后安装所需的软件包: pip install -r requirements.txt 运行SimSiam CUDA_VISIBLE_DEVICES=0 python main.py --data_dir ../Data/ --log_dir ../logs/ -c configs/simsiam_cifar.yaml --ckpt_dir ~/.cache/ --hide_progress