skorch-dev/skorch

Print param groups after initialization

Open

#291 opened on Jul 19, 2018

 (4 comments) (0 reactions) (0 assignees)Jupyter Notebook (414 forks)auto 404
enhancementgood first issue

Repository metrics

Stars
 (6,171 stars)
PR merge metrics
 (PR metrics pending)

Description

We should be verbose about the param groups that are added after initialization (since we support wildcards and it may be intransparent which param groups are created from that).

For example:

>>> net = Net(optimizer__param_groups=[('dense*', {'lr': 0.03})])
Initializing module!
Setting param group {'lr': 0.03} for dense0.weight, dense0.bias, dense1.weight, dense1.bias.

Contributor guide