上传者: 42108054
|
上传时间: 2021-12-12 18:07:43
|
文件大小: 168KB
|
文件类型: -
消息
已经两个月了,我想我终于找到了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