Transfer Learning using ResNet Model

Learn how to train, evaluate, and visualize the performance of a ResNet model.

We train the ResNet model by applying the train_one_epoch function for the desired number of epochs. This is a few epochs since we are fine-tuning the network.

Set up TensorBoard in Flax

To monitor model training via TensorBoard, we can write the training and validation metrics to TensorBoard.

Get hands-on with 1200+ tech skills courses.