[ANN] NNParamsPrinter.jl - Quickly print the layers and weights of your Lux NNs

NNParamsPrinter.jl is a lightweight, simple printer of NN parameters. This is all it does. Please check the GitHub readme for usage.

Why did I make this ?

I wanted to quickly see the weights of a simple NN in a readable way. My use case was in universal differential equations. I had to write a small piece of code for it, which I thought others could benefit from, save them a few minutes :slight_smile:

I hope it is useful in some way.

Thank you!

5 Likes

Cool! Do you think it can be contributed to Lux itself? :slight_smile:

If you use Experimental Features | Lux.jl Docs then you can extend your function to support all layers in a generic fashion

2 Likes

Thanks for the suggestions @Tortar and @avikpal ! I will work on these :slight_smile: