OpenGVLab / UniFormerV2

[ICCV2023] UniFormerV2: Spatiotemporal Learning by Arming Image ViTs with Video UniFormer
https://arxiv.org/abs/2211.09552
Apache License 2.0
294 stars 19 forks source link

Error when performing evaluation/inference [MIT] #16

Closed anjugopinath closed 1 year ago

anjugopinath commented 1 year ago

I am trying to run evaluation.

I set pretrained=False in the function uniformerv2_l14_336 in the file slowfast/models/uniformerv2_model.py

This is the command I executed: ./exp/mit/mit_l14_f8x336/test.sh

But, I am getting this error:

image

Thank You

Andy1621 commented 1 year ago

Since the code is for training, it will load Kinetics pretrained model. You can delete cfg.UNIFORMERV2.PRETRAIN in config.yaml if you only want to inference.

anjugopinath commented 1 year ago

Thanks, that fixed it.