Skip to content

Instantly share code, notes, and snippets.

@tchaton

tchaton/video_clas.py

Created May 11, 2021 11:16
Show Gist options
  • Save tchaton/0e83bff7e27e125cc675bcdd31d3bf23 to your computer and use it in GitHub Desktop.
Save tchaton/0e83bff7e27e125cc675bcdd31d3bf23 to your computer and use it in GitHub Desktop.
Download ZIP
video_clas
Raw
video_clas.py
# 5. Build the VideoClassifier with a PyTorchVideo backbone.
model = VideoClassifier(
backbone="x3d_xs",
num_classes=datamodule.num_classes,
serializer=Labels()
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment