1. 测试模型¶
1.1. 已集成内容¶
目前项目已集成的测试模型共有以下几个:
针对Cifar-10数据集的ResNet20,FP_ResNet,VGG16
针对ImageNet数据集的VGG19
目前项目已集成的模型参数共有以下几个:
resnet20模型在cifar10原始训练集上模型参数
resnet20模型通过RAND加固算法加固后模型参数
FP_ResNet模型在cifar10原始训练集上模型参数
VGG16模型在cifar10原始训练集上模型参数
VGG16模型通过RAND加固算法加固后模型参数
1.2. 自定义模型¶
用户可按照模型扩展要求,实现并上传自定义模型。
也可默认l 使用pytorch自带的模型,以及默认的预训练模型,具体有例如:
torchvision.model.vgg19
torchvision.model.alexnet