search

p-christ / nn_builder

Build neural networks with less boilerplate code
MIT License
162 stars 22 forks source link

Add more network types #9

Open p-christ opened 5 years ago

p-christ commented 5 years ago

At the moment we have NN, CNN and RNN network types but we could add more

e.g. we could add classes that build ResNets, U-Nets, Inception Nets, Autoencoders