Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Fix typo in VideoClassifier docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
fcakyon committed Jun 5, 2022
1 parent 61e1a2d commit a58725d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flash/video/classification/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ class VideoClassifier(ClassificationTask):
Args:
num_classes: Number of classes to classify.
backbone: A string mapped to ``pytorch_video`` backbones or ``nn.Module``, defaults to ``"slowfast_r50"``.
backbone: A string mapped to ``pytorch_video`` backbones or ``nn.Module``, defaults to ``"x3d_xs"``.
backbone_kwargs: Arguments to customize the backbone from PyTorchVideo.
pretrained: Use a pretrained backbone, defaults to ``True``.
loss_fn: Loss function for training, defaults to :func:`torch.nn.functional.cross_entropy`.
Expand Down

0 comments on commit a58725d

Please sign in to comment.