-
使用Tensorflow进行虚拟对抗性训练
资源介绍
vat_tf
用于在SVHN和CIFAR-10数据集上重现半监督学习结果的Tensorflow实现,在论文“虚拟对抗训练:一种用于监督和半监督学习的正则化方法”中
要求
tensorflow-gpu 1.1.0,scipy 0.19.0(用于ZCA美白)
准备半监督学习数据集
在CIFAR-10上
python cifar10.py --data_dir=./dataset/cifar10/
在SVHN上
python svhn.py --data_dir=./dataset/svhn/
无需扩充的半监督学习
在CIFAR-10上
python train_semisup.py --dataset=cifar10 --data_dir=./dataset/cifar10/ --log_dir=./log/cifar10/ --num_epochs=500 --epoch_decay_