文件名称:SLA:通过输入变换进行自我监督的标签增强(ICML 2020)
文件大小:41KB
文件格式:ZIP
更新时间:2024-05-02 21:50:40
Python
通过输入变换进行自我监督的标签增强 作者:,,(KAIST) 接受ICML 2020 安装依赖项 conda create -n SLA python=3.7 conda activate SLA conda install pytorch torchvision cudatoolkit=10.1 -c pytorch conda install ignite -c pytorch pip install tensorboard 我们在以下版本上测试了我们的代码: pytorch==1.5.1 torchvision=0.6.1 ignite==0.4.0.post1 训练 我们在这里提供cifar10的培训脚本。 python train.py \ --dataset cifar10 --datadir data/ --batchsize 128 --num-iter
【文件预览】:
SLA-master
----models()
--------__init__.py(2KB)
--------cifar_resnet.py(6KB)
--------modules.py(805B)
----splits()
--------cifar10_val_idx.npy(39KB)
--------cifar100_val_idx.npy(39KB)
----inat.py(3KB)
----augmentations.py(3KB)
----utils.py(1KB)
----test.py(4KB)
----train.py(7KB)
----datasets.py(6KB)
----trainers.py(6KB)
----.gitignore(18B)
----README.md(2KB)