python train_imagenet.py --ngpu 1 --workers 20 --arch resnet --depth 50 --epochs 100 --batch-size 32 --lr 0.1 --att-type CBAM --prefix RESNET50_IMAGENET_CBAM
cutepkq / cbam_attention-for-stanford_dogs Goto Github PK
View Code? Open in Web Editor NEW用cbam注意力做stanfordDogs分类
License: MIT License