Misleading documentation for freezing of layers

In the documentation for fastai.basic_train.freeze() it is written that the function will “freeze up to last layer”. What actually happens is that the function freezes up to last layer group. The docstring in the source code is likewise misleading.

In the documentation for fastai.basic_train.freeze_to(n:int) it is written that the function will “freeze layers up to layer n”. What actually happens is that the function freezes up to layer group n. The docstring in the source code is likewise misleading.

Correcting this would cause a lot less confusion.

1 Like

Just fo ahead and suggest a PR :wink:
You only need to change the 1-line docstrings in the modules I believe, as those are pulled automatically in the docs.

2 Likes

Yes, of course. I just did.

3 Likes