PyTorch implementation of Res2Net
This repository contains code for Res2Net based on Res2Net: A New Multi-scale Backbone Architecture implemented in PyTorch.
Requirements
- Python 3.6
- PyTorch 1.0
Training
CIFAR-100
- Res2NeXt-29, 6cx24wx4scale:
python train.py --dataset cifar100 --arch res2next29_6cx24wx4scale
- Res2NeXt-29, 8cx25wx4scale:
python train.py --dataset cifar100 --arch res2next29_8cx25wx4scale
- Res2NeXt-29, 6cx24wx6scale:
python train.py --dataset cifar100 --arch res2next29_6cx24wx6scale
- Res2NeXt-29, 6cx24wx4scale-SE:
python train.py --dataset cifar100 --arch res2next29_6cx24wx4scale_se
- Res2NeXt-29, 8cx25wx4scale-SE:
python train.py --dataset cifar100 --arch res2next29_8cx25wx4scale_se
- Res2NeXt-29, 6cx24wx6scale-SE:
python train.py --dataset cifar100 --arch res2next29_6cx24wx6scale_se
ImageNet
- Res2Net-50:
python train.py --dataset imagenet --imagenet-dir </path/to/data> --arch res2net50 --epoch 100 --milestones 30 --weight-decay 1e-4
- Res2Net-101:
python train.py --dataset imagenet --imagenet-dir </path/to/data> --arch res2net101 --epoch 100 --milestones 30 --weight-decay 1e-4
- Res2Net-152:
python train.py --dataset imagenet --imagenet-dir </path/to/data> --arch res2net152 --epoch 100 --milestones 30 --weight-decay 1e-4
- Res2NeXt-50_32x4d:
python train.py --dataset imagenet --imagenet-dir </path/to/data> --arch res2next50_32x4d --epoch 100 --milestones 30 --weight-decay 1e-4
- SE-Res2Net-50:
python train.py --dataset imagenet --imagenet-dir </path/to/data> --arch se_res2net50 --epoch 100 --milestones 30 --weight-decay 1e-4
Results
Coming soon...