Skip to content

pauldragan/optimal-architectures

Repository files navigation

The purpose of this repository is to study numerical optimization techniques and their application in the design of convolutional neural networks.

Currently implemented:

  • assignment of the maximum amount of layer output channels such that a predefined maximum peak memory is not exceeded
    • [] WIP DeepLabV3 segmentation network (need to implement constraints on layers that do not change the number of output channels or require the same input channels at their inputs)

We make use of the tflite-tools repository for tensor lifeness and scheduling analysis.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published