Thanks again for the nice comments!
May I ask one more quick question? How do I freeze only a subset of the variables during the training process?
For example, if I have a layer with weight matrix W and bias vector b, how do I only freeze b and still update the W during the training process?