deephealthproject / eddl

European Distributed Deep Learning (EDDL) library. A general-purpose library initially developed to cover deep learning needs in healthcare use cases within the DeepHealth project.
https://deephealthproject.github.io/eddl/
MIT License
34 stars 10 forks source link

Implement LRN #310

Open salvacarrion opened 2 years ago

salvacarrion commented 2 years ago

This layer is used in many early DNNs such as AlexNet, GoogLeNet, Inception, etc. so we need to implement it to support these models through ONNX.

For now, we have two workarounds to be made:

Info:

salvacarrion commented 2 years ago

I think it is supported by cuDNN: https://docs.nvidia.com/deeplearning/cudnn/api/index.html#cudnnSetLRNDescriptor