facebookresearch / TimeSformer

The official pytorch implementation of our paper "Is Space-Time Attention All You Need for Video Understanding?"
Other
1.55k stars 212 forks source link

question #97

Closed hbrachemi closed 2 years ago

hbrachemi commented 2 years ago

I would like to edit the model to train on a regression task rather than a classification using the available weights