Skip to content

XiaobingSuper/convnet-benchmark-py

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Convnet Performance Benchmark for PyTorch

usage

### the script runs on GPU when GPU is available, otherwise on CPU
### for training performance (batched mode)
./run.sh
### for inference performance (batched mode)
./run.sh --inference
### for inference latency (single batch)
./run.sh --inference --single

### for training performance (batched mode)
./run.sh --mkldnn
### for inference performance using mkldnn tensor (batched mode)
./run.sh --inference --mkldnn
### for inference latency using mkldnn tensor (single batch)
./run.sh --inference --single --mkldnn

About

PyTorch convnet performance benchmark

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 97.8%
  • Shell 2.2%