为了解决这个问题,需要升级Tensorflow 2.3.0或更高版本,并且在Bert模型的编译代码中删除steps_per_execution
关键字参数。例如,以下代码片段中的steps_per_execution
可以被删除:
model.compile(optimizer=optimizer,
loss=loss_fn,
metrics=[metric_fn],
steps_per_execution=10)
通过删除关键字参数,Bert模型就可以成功编译并运行。