神经网络的随机坐标下降
指导
目录
main包含具有不同神经网络的scd的培训脚本。
core包含网络结构,损耗函数和优化的定义。
tools包含Args定义,数据集准备和加载界面以及其他工具将在训练脚本中使用。
Pytorch_CIFAR10 PGD攻击代码。
Python环境
python v3.6.8
pytorch v1.7
的numpy v1.17.4
熊猫v1.0.3
argparse v1.1
sklearn v0.21.2
advertorch v0.2( )
ART v1.4.0( )