Skip to content

Comments

Add lr_mult label to the network graph in draw_net.py#6273

Merged
Noiredd merged 2 commits intoBVLC:masterfrom
nic25:draw
Mar 7, 2018
Merged

Add lr_mult label to the network graph in draw_net.py#6273
Noiredd merged 2 commits intoBVLC:masterfrom
nic25:draw

Conversation

@nic25
Copy link
Contributor

@nic25 nic25 commented Mar 3, 2018

adds an option to visualize information about the learning rate (multipliers) when drawing a networks graph.
#6244 @Noiredd

rewrote the function body to make it more streamlined.
does not affect inputs and outputs
@Noiredd Noiredd changed the title Draw Add lr_mult label to the network graph in draw_net.py Mar 5, 2018
Copy link
Member

@Noiredd Noiredd left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me, tested on a few nets of my own. Please fix this one little issue in the comments, and I'll merge.

Parameters
----------
layer : caffe_pb2.LayerParameter
A Convolu--display_lrmtion, Deconvolution, or InnerProduct layer.
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Accidental middle-click? ;)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yeah :-)

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.
@nic25
Copy link
Contributor Author

nic25 commented Mar 6, 2018

fixed, thanks!

@Noiredd
Copy link
Member

Noiredd commented Mar 7, 2018

Great, thank you for the contribution!

@Noiredd Noiredd merged commit 7b3ac40 into BVLC:master Mar 7, 2018
oscarriddle pushed a commit to oscarriddle/caffe that referenced this pull request Mar 18, 2018
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.
XinYao1994 pushed a commit to XinYao1994/caffe that referenced this pull request Aug 29, 2018
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.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants