Orbit is a customized training loop library built on top of Tensorflow 2. It
provides a flexible lightweight library that users can easily use or fork when
writing customized training loop code
in TF2. It intergates with tf.distribute
seamlessly and supports running on
different device types (CPU, GPU, and TPU).
Files
Latest commit
This branch is 3817 commits behind tensorflow/models:master.
orbit
Folders and files
Name | Name | Last commit date | ||
---|---|---|---|---|
parent directory.. | ||||