How to convert TimeSformer Implementation For Regression Tasks

mx-mark / VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.

272 stars 34 forks source link

How to convert TimeSformer Implementation For Regression Tasks #30

Open Taimoor-R opened 1 year ago

Taimoor-R commented 1 year ago

Hi, Great work! I was just wondering how to modify the implemented code for a regression task. Furthermore, can we use custom dataloader for the datasets as the dataset I am using is frames and thier corresponding values(regression labels).