Bert文本分类run_classifier的预测模块修改
修改位置1:run_classifier.py model_fn() 函数中:
源码1:
else:
output_spec = tf.contrib.tpu.TPUEstimatorSpec(
mode=mode, predictions=probabilities, scaffold_fn=scaffold_fn)
替换源码1:
elif mode == tf.estimator.Mo...
原创
2019-08-20 00:18:24 ·
1568 阅读 ·
2 评论