### 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
forked from mingfeima/convnet-benchmark-py
-
Notifications
You must be signed in to change notification settings - Fork 0
XiaobingSuper/convnet-benchmark-py
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
PyTorch convnet performance benchmark
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 97.8%
- Shell 2.2%