Allocate the specified parameters to optimize
Created by: CharlesxrWu
I face the problem that I didn't find the API which support to optimize the specified parameters. For example, in the trainer.SGD class, if the parameters are the part of the parameters instead of the whole parameter dictionary, will we just optimize the specified parameters to reduce the loss?
Something like var list in tensorflow optimizer.minimize(, var_list=), just optimize the parameters in the list.