Code for Deep Learning frameworks benchmark. Currently work in progress, basic benchmark with artifically generated data was made.
keras/
contains code for Keras benchmarkingpytorch/
contains code for Pytorch benchmarkingscripts/schedule.sh
runs set of models from both benchmarksscripts/log_extract.ipynb
notebook with logs data processing & savingscripts/*.R
R scripts for data visualization based on processed logs
NASNet model wasn't working with tensorflow.keras
. It threw errors, so it was finally omitted.
Versions:
- TF: 1.12.0
- Keras: 2.2.4
- Torch: 1.0.0