Summary: TensorBoard and State Handling
Let’s summarize the concepts that we have learned so far.
We'll cover the following...
Recap
In this chapter, we saw how we can use TensorBoard to log our experiments in Flax. We have also seen how to build networks in Flax containing the BatchNorm and DropOut ...