
ResNet_Attention: ResNet结合注意力机制。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
ResNet_Attention(CBAM,SE)的官方文档指出,运行该模型所需的操作系统环境为Ubuntu 20.04,并建议使用GTX 1080Ti显卡以及Python 3.7作为编程语言。此外,还需要安装PyTorch 1.7.0和CUDA 10.2,以及CuDNN 7.0。该模型的核心架构是ResNet。在我们的实验过程中,我们选择CIFAR10数据集进行训练。具体而言,为了使用SE模块进行训练,我们执行了 `python train_CIFAR10.py --prefix 4 --device 1 --epoch 160 --att_type se` 命令;而为了使用CBAM模块进行训练,则执行了 `python train_CIFAR10.py --prefix 5 --device 1 --epoch 160 --att_type cbam` 命令。验证结果显示,在经过160个时代训练的ResNet50模型中,ACC@1达到了93.41%。
全部评论 (0)
还没有任何评论哟~


