vc.py

# Load the data from directories
datamodule = VideoClassificationData.from_folders(
    train_folder="./train",
    clip_sampler="uniform",
    clip_duration=1,
    video_sampler=RandomSampler,
    decode_audio=False)

# Build the VideoClassifier with a PyTorchVideo backbone.
model = VideoClassifier(
    backbone="x3d_xs",
    num_classes=datamodule.num_classes, 
    serializer=Labels())

# Finetune the model
trainer = flash.Trainer()
trainer.finetune(model, datamodule=datamodule)

# Make a prediction
model.predict("./videos_dir")