• Hongjun Choi's avatar
    Merged commit includes the following changes: (#6921) · d76e39e7
    Hongjun Choi authored
    250606180  by A. Unique TensorFlower<gardener@tensorflow.org>:
    
        Fix BERT benchamrk test errors.
    
    --
    250589623  by A. Unique TensorFlower<gardener@tensorflow.org>:
    
        Change BERT benchmark test pretrained checkpoint url.
    
    --
    250587892  by A. Unique TensorFlower<gardener@tensorflow.org>:
    
        Fix error in BERT custom training loop checkpoint restoration.
    
    --
    250577163  by A. Unique TensorFlower<gardener@tensorflow.org>:
    
        Add logic to inject callback that measures performance in BERT custom training
        loop.
    
    --
    250529526  by hongkuny<hongkuny@google.com>:
    
        Internal clean up
    
    --
    250428976  by hongkuny<hongkuny@google.com>:
    
        Internal change
    
    250415383  by A. Unique TensorFlower<gardener@tensorflow.org>:
    
        Add min/max value to BERT classifier benchmark test.
    
    --
    250376246  by A. Unique TensorFlower<gardener@tensorflow.org>:
    
        Add benchmark performance test to run BERT on multiple numbers of GPUs.
    
    --
    
    PiperOrigin-RevId: 250606180
    d76e39e7
model_training_utils.py 11 KB