Skip to content

Fix LayerTag docstring: remove misleading 'with the given name'#403

Open
Muneerali199 wants to merge 1 commit into
google-deepmind:mainfrom
Muneerali199:fix/layertag-docstring
Open

Fix LayerTag docstring: remove misleading 'with the given name'#403
Muneerali199 wants to merge 1 commit into
google-deepmind:mainfrom
Muneerali199:fix/layertag-docstring

Conversation

@Muneerali199

Copy link
Copy Markdown

Fixes #401

The __init__ docstring for LayerTag said it initializes a primitive
"with the given name", implying a name parameter. The constructor takes
no arguments and hardcodes the name to "layer_tag".

This caused confusion for users who tried to pass a name argument (e.g.,
LayerTag("my_tag")).

The __init__ docstring for LayerTag said it initializes 'with the given
name', but the constructor takes no arguments and hardcodes the name to
'layer_tag'. This caused confusion for users who tried to pass a name
argument. Remove the misleading phrase.
@brockdyer03

Copy link
Copy Markdown

It'd be nice to know how to migrate old code to the new system. I am trying to update some code that had

import kfac_jax
repeated_dense_tag = kfac_jax.LayerTag("repeated_dense_tag", 1, 1)

to use the new system, but I'm unfamiliar with the correct way to register a tag now.

It's especially difficult since the documentation has not been re-built in over 2 years.

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.

LayerTag documentation not consistent with LayerTag class

2 participants