使用swin-transformer里的predict.py和官方swin_tiny_patch4_window7_224.pth模型预测没结果；

[LegendSun0]

System information

• Have I written custom code:
• OS Platform(e.g., window10 or Linux Ubuntu 16.04):
• Python version:
• Deep learning framework and version(e.g., Tensorflow2.1 or Pytorch1.3):
• Use GPU or not:
• CUDA/cuDNN version(if you use GPU):
• The network you trained(e.g., Resnet34 network):

Describe the current behavior

Error info / logs

[LegendSun0]

使用官方swin_tiny_patch4_window7_224.pth模型预测没有报错，但返回的结果都是0。请问是什么问题？

[WZMIAOMIAO]

没理解你说的问题，使用predict脚本会打印每个类别的名称和概率，全0是什么意思