Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Parameter description error: pnet_tune #131

Open
pbhfcycssjlmm opened this issue Apr 26, 2024 · 0 comments
Open

Parameter description error: pnet_tune #131

pbhfcycssjlmm opened this issue Apr 26, 2024 · 0 comments

Comments

@pbhfcycssjlmm
Copy link

In your description of the pnet_tune parameter:
pnet_tune
[False] tune the base/trunk network
[True] keep base/trunk frozen
However:
self.net = net_type(pretrained=not self.pnet_rand, requires_grad=self.pnet_tune)
Thus,the implementation in the code shows:
When pnet_tune=True, set by requires_grad=self.pnet_tune, it means that the weights of the base network will be updated (that is, participate in gradient calculation).
When pnet_tune=False, the weights of the base network will be frozen and will not participate in gradient updates.
So the correct comment should be:
pnet_tune=True: Indicates to "unfreeze" the base network, allowing its weights to be updated during training.
pnet_tune=False: means to "freeze" the base network and its weights will not be updated during training.

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

No branches or pull requests

1 participant