I\'m trying to use transfer learning to solve a problem, but when I insert regularization term in pre-trained model and I try to use it in train_step, I have th
train_step