If we load pre-trained weight and do some training, after that, when saving weights will not save updated weights. It's because we didn't get the latest weights (_arg_params) in bucketing module.
fixed with new get_arg_param method when saving weights
Summary
fix https://github.com/awslabs/keras-apache-mxnet/issues/138
If we load pre-trained weight and do some training, after that, when saving weights will not save updated weights. It's because we didn't get the latest weights (_arg_params) in bucketing module.
fixed with new get_arg_param method when saving weights
added unit test for testing fine-tune use case.
Related Issues
https://github.com/awslabs/keras-apache-mxnet/issues/138
PR Overview