SkywardAI / kimchima

Customise neural nets, knowledge distillation and transfer learning collection
https://pypi.org/project/kimchima/
Apache License 2.0
0 stars 2 forks source link

Add MLP with batchNorm layer #123

Closed Aisuko closed 3 months ago

Aisuko commented 3 months ago

Similar implement solution with https://www.kaggle.com/code/aisuko/implement-neural-net-with-batch-norm-layer

Batch Norm architecture follows https://arxiv.org/abs/1502.03167