Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Register all the parameters in the optimizer (pytorch#1455)
The code at the end registers only the parameters from `model.fc` in the optimizer, although the text underneath says: "Notice although we register all the parameters in the optimizer, the only parameters that are computing gradients (and hence updated in gradient descent) are the weights and bias of the classifier." To be consistent with this explanation, we should be adding all the parameters from the model.
- Loading branch information