facebookresearch / TimeSformer

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

How to load pretrained SlowFast Model? #116

Open HashmatShadab opened 2 years ago

HashmatShadab commented 2 years ago

for loading the slow fast models, cfg.TASK is required while building the model. However, it is not defined anywhere in the config file. To load the pretrained models, I am using weights from https://github.com/facebookresearch/SlowFast/blob/main/MODEL_ZOO.md. For them to be loaded properly,, cfg.TEST.CHECKPOINT_TYPE == "caffe2". Can you please let me know if this is the correct way of loading the model?