SLA_CHECK:查看 源码

通过输入变换进行自我监督的标签增强 作者:,,(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