Installations and Imports

Hyperparameters

Dataset and Dataloaders

Compile the Model

We will use CosineDecay learning rate scheduler.

[OPTIONAL] Model Prediction Visualization

We will build a custom Keras callback by subclassing WandbEvalCallback for model prediction visualization.

Train the model with W&B

Model Evaluation