tensorflow estimator 使用hook实现finetune方式

为了实现finetune有如下两种解决方案： model_fn里面定义好模型之后直接赋值 def model_fn(features, labels, mode, params): # ..... # finetune if params.checkpoint_path and (not tf.train.latest_checkpoint(params.model_dir)): checkpoint_path = None if tf.gfile.IsDirectory(params.checkpoint_path): checkpoint_path = tf.train.