GitLab

Testing: Trained Bert2Bert model with MLIR bridge enabled and then NHNet based on Bert2Bert checkpoint, also with MLIR bridge enabled. Exported this to a saved model, ran inference, and confirmed model performance, accuracy, and prediction parity with the model trained without the MLIR bridge enabled.
PiperOrigin-RevId: 326734770

PiperOrigin-RevId: 326731205

PiperOrigin-RevId: 326722720

i.e., the checkpoint will be saved using
tf.train.Checkpoint(FLAGS.checkpoint_model_name=model)

PiperOrigin-RevId: 326672697

PiperOrigin-RevId: 326604435

PiperOrigin-RevId: 326586473

PiperOrigin-RevId: 326541553

PiperOrigin-RevId: 326534159

PiperOrigin-RevId: 326496940

PiperOrigin-RevId: 326396920

PiperOrigin-RevId: 326286926

Fix QQP preprocessing for test set. Test set does not contain qid1 or qid2 columns, so the current try/except block causes the entire test set to be ignored.

PiperOrigin-RevId: 326259054

PiperOrigin-RevId: 326246781

Project name seems that have changed to project ID. Just changed the project names to project Id.

Add ops for extracting labels from sentences (get_*_labels() functions). This CL includes the ops for extracting labels for:
- BERT's next sentence prediction task
- ALBERT's sentence order prediction task

PiperOrigin-RevId: 326087418

PiperOrigin-RevId: 326068206

PiperOrigin-RevId: 325956876

PiperOrigin-RevId: 325736862

PiperOrigin-RevId: 325713616

PiperOrigin-RevId: 325490166

The example config has b.ccc.a1 = 1.

PiperOrigin-RevId: 325478817

PiperOrigin-RevId: 325454896