Skip to content

Files

orbit

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Apr 10, 2021
Mar 18, 2021
Jul 9, 2020
Nov 19, 2020
Jan 5, 2021
Feb 5, 2021
Jan 5, 2021
Jan 5, 2021
Mar 30, 2021
Mar 22, 2021

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.