比如,只训练一阶和二阶的特征向量。
亲测,当冻结某些层,训练层时效果会提升不少。
train_var_list = list()
for k in tf.global_variables():
if k.name.startswith("one_order") or k.name.startswith("two_order"):
train_var_list.append(k)
train_op = tf.train.GradientDescentOptimizer(LEARNING_RATE).minimize(
train_loss, global_step=global_step, var_list=train_var_list)