Error when importing flash.video in v0.8.1
DManowitz opened this issue · comments
🐛 Bug
When trying to import flash.video from lightning-flash 0.8.1, I get this error:
╭─────────────────────────────── Traceback (most recent call last) ────────────────────────────────╮
│ in <module> │
│ │
│ C:\Users\manow\miniconda3\envs\py38_ml_TfGpuMain\lib\site-packages\flash\video\__init__.py:2 in │
│ <module> │
│ │
│ 1 from flash.video.classification.data import VideoClassificationData # noqa: F401 │
│ ❱ 2 from flash.video.classification.model import VideoClassifier # noqa: F401 │
│ 3 │
│ │
│ C:\Users\manow\miniconda3\envs\py38_ml_TfGpuMain\lib\site-packages\flash\video\classification\mo │
│ del.py:44 in <module> │
│ │
│ 41 │ │ │ │ _VIDEO_CLASSIFIER_BACKBONES(fn=fn, providers=_PYTORCHVIDEO) │
│ 42 │
│ 43 │
│ ❱ 44 class VideoClassifier(ClassificationTask): │
│ 45 │ """Task that classifies videos. │
│ 46 │ │
│ 47 │ Args: │
│ │
│ C:\Users\manow\miniconda3\envs\py38_ml_TfGpuMain\lib\site-packages\flash\video\classification\mo │
│ del.py:81 in VideoClassifier │
│ │
│ 78 │ │ loss_fn: LOSS_FN_TYPE = F.cross_entropy, │
│ 79 │ │ optimizer: OPTIMIZER_TYPE = "Adam", │
│ 80 │ │ lr_scheduler: LR_SCHEDULER_TYPE = None, │
│ ❱ 81 │ │ metrics: METRICS_TYPE = Accuracy(), │
│ 82 │ │ learning_rate: Optional[float] = None, │
│ 83 │ │ head: Optional[Union[FunctionType, nn.Module]] = None, │
│ 84 │ ): │
╰──────────────────────────────────────────────────────────────────────────────────────────────────╯
TypeError: __new__() missing 1 required positional argument: 'task'
To Reproduce
import flash.video
Code sample
See above
Expected behavior
Not an error.
Environment
- OS (e.g., Linux): Windows 10
- Python version: 3.8
- PyTorch/Lightning/Flash Version (e.g., 1.10/1.5/0.7): 1.12.1 / 1.6.5 / 0.8.1
- GPU models and configuration: NVIDIA GeForce RTX 3060 Laptop GPU
- Any other relevant information:
Additional context
N/A
@DManowitz would need to downgrade torchmetrics version...
or mind sending a fix with setting this argument for Accuracy? 🐰
seems to be working fine on master, so pls install from source
from flash.video.classification.model import VideoClassifier