ml-performance-prediction

KO | EN

gitlite — search

#typescript #golang #go #rust #deep-learning #linux #java #cli #kubernetes #awesome-list #vue #pytorch #security #ios

ml-performance-prediction

(★ 21)

Code that accompanies the paper "Predicting the Computational Cost of Deep Learning Models"

ml-performance-prediction Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# performance-prediction
Code related to the paper **Predicting the Computational Cost of Deep Learning Models**. This code allows to train a machine learning model that can predict the execution time for commonly used layers within deep neural networks - and, by combining these, for the full network.

A python package that utilises this model for inference can be found at https://github.com/CDECatapult/mlpredict.

This work is intended as starting point for an open source machine learning tool, capable of accurately predicting the time that is required to train any neural network on any given hardware. As such, it is easy for everyone to add additional hardware, model layers, or input features, or optimise the prediction model itself.

The folder *benchmark* contains code for benchmarking deep neural networks as well as single layers within these.

The folder *prediction_model* contains code to generate training data for the model described in the above paper, a data preparation pipeline, and the model training procedures. This folder also contains the training data and the existing tensorflow models.

ml-performance-prediction

ml-performance-prediction Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

ml-performance-prediction

ml-performance-prediction Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README