过滤器相似度:通过降低过滤器间相似度来提高各种CNN的性能

时间:2024-03-18 05:16:52
【文件属性】:

文件名称:过滤器相似度:通过降低过滤器间相似度来提高各种CNN的性能

文件大小:46KB

文件格式:ZIP

更新时间:2024-03-18 05:16:52

Python

过滤器相似度 CIFAR10模型砝码 使用链接下载权重(您必须手动下载和提取)。 要将路径设置为下载的weights文件夹,请在vgg.py指定path_to_state_dict 。 依存关系 通过运行以下命令安装依赖项: pip install -r requirements.txt 选项: -net -架构的选择(默认:resnet18) -dataset选择数据集(默认值:cifar10) -tr-batch训练批次大小(默认值:512) -val-batch验证批次大小(默认值:512) -lr学习率(默认值:0.1) -wd权重衰减(默认值:5e-4) -epochs要训​​练的时期数(默认值:300) -cpu -cpu标志-reinit分集丢失使用标志-mode -在转移和默认训练之间选择 如何运行: python train.py -lr 0.1 -gpu -dat


【文件预览】:
filter-similarity-main
----.gitignore(2KB)
----requirements.txt(51B)
----black.toml(524B)
----models()
--------xception.py(6KB)
--------shufflenetv2.py(5KB)
--------vgg.py(2KB)
--------mobilenet.py(4KB)
--------inceptionv3.py(11KB)
--------shufflenet.py(7KB)
--------senet.py(5KB)
--------resnext.py(4KB)
--------squeezenet.py(2KB)
--------nasnet.py(9KB)
--------rir.py(7KB)
--------densenet.py(5KB)
--------attention.py(11KB)
--------preactresnet.py(4KB)
--------googlenet.py(4KB)
--------mobilenetv2.py(3KB)
--------inceptionv4.py(17KB)
--------resnet.py(5KB)
----utils.py(9KB)
----README.md(2KB)
----test.py(3KB)
----dataset.py(5KB)
----conf()
--------__init__.py(300B)
--------global_settings.py(1KB)
----main.py(13KB)
----train.py(13KB)

网友评论