awslabs / keras-apache-mxnet

[DEPRECATED] Amazon Deep Learning's Keras with Apache MXNet support
https://github.com/awslabs/keras-apache-mxnet/wiki
Other
290 stars 65 forks source link

fix kwargs and add comments #254

Closed roywei closed 4 years ago

roywei commented 4 years ago

Summary

  1. Fix kwargs In https://github.com/awslabs/keras-apache-mxnet/pull/250, I added **kwargs in _init_subclassed_network inorder to fix MXNet context problem, and it's not enough.

This PR brings more **kwargs in Keras front end to fix **kwargs not passed along in subsequent _base_init method.

These changed are already made in official upstream Keras front end, and will be updated during next merge.

  1. Updated comments to properly fix moving mean and var for BatchNorm layer

Related Issues

PR Overview