Add lr_mult label to the network graph in draw_net.py (#6273)
authornic25 <nicola.corradi@gmail.com>
Wed, 7 Mar 2018 11:40:52 +0000 (03:40 -0800)
committerWook Song <wook16.song@samsung.com>
Thu, 23 Jan 2020 13:50:35 +0000 (22:50 +0900)
commit7bc7b007273e58ef7407e9e9a1062f0f3b1bcf54
treef1dc5c8506a3d7b1fe4c78572b162f15fa0bd80a
parent379a3ba2d5421a6ac05afa4239c30739cc79f7b0
Add lr_mult label to the network graph in draw_net.py (#6273)

draw_net.py refactoring and optional LR visualization

* refactoring `get_layer_label`

rewrote the function body to make it more streamlined.
does not affect inputs and outputs

* optionally visualize LR when drawing the network

adds an option to `python/draw_net.py` that allows to visualize information
about the learning rate multiplier (if relevant) when drawing the network's
graph.
python/caffe/draw.py
python/draw_net.py