ModelBuilder

[dmccloskey]

Description

Class to automate the construction of networks. Would require a separate ModuleBuilder class to define reusable sub units.

Objectives

• [x] FC layer
• number of nodes
• [x] Convolution Layer
• Depth (# of filters), stride (spacing between filters), spatial extent(width and height), zero padding
• [x] Pooling layer (max)
• Spatial extent (width and depth), stride
• [x] LSTM/GRU layer
• number of hidden units, architecture
• [x] SoftMax
• number of nodes
• [ ] Custom module