使用tf.contrib.tpu.TPUEstimatorSpec 进行bert finetune 训练时，无法输出训练过程的loss

作者：小蓝xlanll | 2024-03-25 20:32:12

踩

tf.contrib.tpu.tpuestimatorspec

使用 logging_hook = tf.train.LoggingTensorHook({"loss":total_loss},every_n_iter=100)进行显示


if mode == tf.estimator.ModeKeys.TRAIN:
            train_op = optimization.create_optimizer(
                total_loss, learning_rate, num_train_steps, num_warmup_steps, use_tpu)
            logging_hook = tf.train.LoggingTensorHook({"loss":total_loss},every_n_iter=100)
            output_spec = tf.contrib.tpu.TPUEstimatorSpec(
                mode=mode,
                loss=total_loss,
                train_op=train_op,
                training_hooks=[logging_hook],
                scaffold_fn=scaffold_fn)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/312313