Skip to content

4uiiurz1/pytorch-res2net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

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

Releases

No releases published

Packages

No packages published

Languages