Training keras model with pytorch lightning API

[innat]

Due to the support to torch backend in keras 3, it's now possible to run keras model using lightning api. I have tested a video model for classification task. It works just great. Thanks to the lighning team to build such wonderful library.

Notebook: Video Classification

import keras
from keras import ops
import pytorch_lightning as pl
from keras_cv.models import VideoClassifier


keras_model = VideoClassifier()
...
trainer = pl.Trainer(
    accelerator='auto', 
    max_epochs=10, 
    accumulate_grad_batches=2,
    limit_train_batches=len(train_df) // batch_size,
    precision=16,
) 
tmodel = LightningVideoClassifier(keras_model)
trainer.fit(
    tmodel, 
    train_ds,
)