tensorflow / swift

Swift for TensorFlow
https://tensorflow.org/swift
Apache License 2.0
6.13k stars 606 forks source link

Adding a guide to training loops and callbacks #612

Closed BradLarson closed 3 years ago

BradLarson commented 3 years ago

This adds a guide for our experimental training loop abstraction, the built-in callbacks, and how to write a custom callback.

sgugger commented 3 years ago

Oh, thanks for the shout-out!

BradLarson commented 3 years ago

@sgugger - Well, it is your design, so to do anything else would at best be disingenuous. I also made sure to link to your awesome WordSeg analysis in our new guides to hopefully raise the visibility of that as well.