PiperOrigin-RevId: 530426973

PiperOrigin-RevId: 530385565

This CL adds a field in Orbit trainer/controller indicating whether async checkpoint is enabled for checkpoint saving. BY default this value is set to False, which is equivalent to the existing behavior.

In addition, a sync barrier is added at the end of training (in controller) to make sure users code won't prematurely access the checkpoint file/state when the async checkpoint saving is still ongoing.

PiperOrigin-RevId: 529300903

PiperOrigin-RevId: 529212785

PiperOrigin-RevId: 529166220

PiperOrigin-RevId: 528925617

PiperOrigin-RevId: 528905375

PiperOrigin-RevId: 528817322

PiperOrigin-RevId: 528558530

PiperOrigin-RevId: 528555834

PiperOrigin-RevId: 528500682

PiperOrigin-RevId: 528210978

PiperOrigin-RevId: 528061912

PiperOrigin-RevId: 527977345

PiperOrigin-RevId: 527722635

PiperOrigin-RevId: 527614455

PiperOrigin-RevId: 527373279

PiperOrigin-RevId: 527372244

PiperOrigin-RevId: 526625272

PiperOrigin-RevId: 526170026

PiperOrigin-RevId: 526161135

PiperOrigin-RevId: 526151993

Drop TF addons dependency from vision legacy folder. Copy Lookahead optimizer from TF addons to model garden.

PiperOrigin-RevId: 526149719

Remove Deprecated `tf.keras.layers.experimental.SyncBatchNormalization` from existing code and update `tf.keras.layers.experimental.BatchNormalization` with `synchronized` argument.

PiperOrigin-RevId: 526102034

PiperOrigin-RevId: 525618180

PiperOrigin-RevId: 525585347

PiperOrigin-RevId: 525498900

PiperOrigin-RevId: 525291100

PiperOrigin-RevId: 525240094

PiperOrigin-RevId: 525237496

PiperOrigin-RevId: 525208989

PiperOrigin-RevId: 525208722

PiperOrigin-RevId: 524957164

PiperOrigin-RevId: 524906464

PiperOrigin-RevId: 524902271

PiperOrigin-RevId: 524901616

PiperOrigin-RevId: 524770642

PiperOrigin-RevId: 524396218

Add `num_convs` and `num_filters` in attribute head for retinanet so that each attribute head can use different conv settings.

PiperOrigin-RevId: 524181606

PiperOrigin-RevId: 524138987