White Box 源码
白盒 在白盒中执行CNN频道修剪 要求: python3.7.4,pytorch 1.5.1,torchvision 0.4.2,thop 0.0.31 运行以下脚本以重现纸质报告的结果(在相应的脚本中更改数据路径)。 VGGNet-16-CIFAR10 ./scripts/vgg.sh ResNet-56-CIFAR10 ./scripts/resnet56.sh ResNet-110-CIFAR10 ./scripts/resnet110.sh MobileNet-v2-CIFAR10 ./scripts/mobilenetv2.sh ResNet-50-ImageNet(FLOPs:2.22B)./scripts/resnet50-1.sh ResNet-50-ImageNet(FLOPs:1.50B)./scripts/resnet50-2.sh 运行以下脚本来测
文件列表
White-Box-main.zip
(预估有个23文件)
White-Box-main
test_pruned_model.py
6KB
cifar.py
26KB
data
imagenet.py
2KB
cifar10.py
1KB
model
resnet.py
13KB
mobilenet_v2.py
10KB
vgg_cifar.py
9KB
暂无评论