比如,只训练一阶和二阶的特征向量。

亲测,当冻结某些层,训练层时效果会提升不少。

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)