I have implemented the following model. The model has 3 layers and each layer has 15 modules. The modules are constructed of simple fully connected layers. The model takes a
