Skip to content

Latest commit

 

History

History
 
 

orbit

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.