Skip to content
This repository has been archived by the owner on Jul 11, 2022. It is now read-only.
/ pytorch_shake_shake Public archive

A PyTorch implementation of shake-shake

License

Notifications You must be signed in to change notification settings

hysts/pytorch_shake_shake

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch Implementation of shake-shake

Usage

$ python train.py --depth 26 --base_channels 32 --shake_forward True --shake_backward True --shake_image True --outdir results

Results on CIFAR-10

Model Test Error (median of 3 runs) Test Error (in paper) Training Time
shake-shake-26 2x32d (S-S-I) 3.68 3.55 (average of 3 runs) 11h16m
shake-shake-26 2x64d (S-S-I) 2.88 (1 run) 2.98 (average of 3 runs) 19h43m
shake-shake-26 2x96d (S-S-I) 2.90 (1 run) 2.86 (average of 5 runs) 32h10m

Notes

  • Tesla V100 was used in these experiments.

References

About

A PyTorch implementation of shake-shake

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages