【文件属性】:
文件名称:Meta_Pseudo_Labels:“元伪标签”的PyTorch实现
文件大小:220KB
文件格式:ZIP
更新时间:2021-02-04 12:51:56
docker latex imagenet pytorch-implementation pytorch-ignite
元伪标签
安装套件
pip3 install SOTA-SSL
用法
import torch
from SOTA_SSL_Models import SimSiam
from torchvision import models
model = SimSiam(args)
笔记
我发现直接使用SimCLR增强有时会导致模型崩溃。 这可能是由于SimCLR增强太强的事实。
在预热阶段采用MoCo增强会有所帮助。
数据集
data/
imagenet/
train/
...
n021015556/
..
n021015556_