Skip to content

4uiiurz1/pytorch-res2net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

073efde · Apr 28, 2019

History

4 Commits
Apr 8, 2019
Apr 8, 2019
Apr 28, 2019
Apr 8, 2019
Apr 8, 2019

Repository files navigation

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...

About

PyTorch implementation of Res2Net

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages