Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Lucy Fox's avatar
    Enable MLIR bridge by default for NHNet and Bert2Bert training. · 26fc1410
    Lucy Fox authored 
    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
    26fc1410
trainer.py 8.75 KB