Hi, thank you for your reply. When tuning BERT, the error is raised exactly when it tries to tune Batch_MatMul. And I just tried your solution and it worked! Really appreciate you for your help!
Hi, thank you for your reply. When tuning BERT, the error is raised exactly when it tries to tune Batch_MatMul. And I just tried your solution and it worked! Really appreciate you for your help!