BMXNet-v2 Beta
[0.2.0] - 2018-12-04
First beta version.
Note that binary (and quantized) examples are maintained separately in a submodule:
example/bmxnet-examples
Added
- Functions:
- New operator:
contrib.gradcancel
(see src/operator/contrib/gradient_cancel[-inl.h|.cc|.cu])- allows to cancel gradients (element-wise) if absolute value of input is larger than a threshold
- Binary versions of the following layers of the gluon API:
- gluon.nn.Dense -> gluon.nn.QDense
- gluon.nn.Conv1D -> gluon.nn.QConv1D
- gluon.nn.Conv2D -> gluon.nn.QConv2D
- gluon.nn.Conv3D -> gluon.nn.QConv3D
- Tests are in tests/python/unittest/test_binary.py
- Layers are in python/mxnet/gluon/nn/binary_layers.py
Changed
- Code in python/mxnet/visualization.py changed:
plot_network
skips certain layers which clutter binary network graphsprint_summary
calculates compressed model size and prints number of binarized/quantized and full-precision weights