Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 673 Bytes

README.md

File metadata and controls

13 lines (10 loc) · 673 Bytes

Orbit

Orbit is a flexible, lightweight library designed to make it easy to write custom training loops in TensorFlow 2. Orbit handles common model training tasks such as saving checkpoints, running model evaluations, and setting up summary writing, while giving users full control over implementing the inner training loop. It integrates with tf.distribute seamlessly and supports running on different device types (CPU, GPU, and TPU). The core code is intended to be easy to read and fork.

See our g3doc at go/orbit-trainer for additional documentation.