Skip to content

maple-research-lab/CapProNet-Pytorch

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
models
models
 
 
pretrained_model
pretrained_model
 
 
view
view
 
 
README.md
README.md
 
 
extract_features.py
extract_features.py
 
 
logger.py
logger.py
 
 
main.py
main.py
 
 
utils.py
utils.py
 
 

Repository files navigation

CUDA_VISIBLE_DEVICES=0 python main.py $PATH_TO_CIFAR10$ --dataset cifar10 --arch resnet110 --save_path ./110epoch500D8 --epochs 500 --schedule 250 375 --gammas 0.1 0.1 --learning_rate 0.1 --decay 0.0001 --batch_size 128 --Ddim 8

pre-trained model: ./pretrained_model

About

Capsule Projection Networks (CapProNet) in pytorch, NeurIPS 2018

arxiv.org/abs/1701.06264

Resources

Readme
Activity
Custom properties

Stars

33 stars

Watchers

2 watching

Forks

10 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages