search

mit-han-lab / temporal-shift-module

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding
https://arxiv.org/abs/1811.08383
MIT License
2.05k stars 418 forks source link

模型输出类别固定 #214

Closed dengfenglai321 closed 2 years ago

dengfenglai321 commented 2 years ago

你好， 我训练4类别数据， 不过发现模型的固定输出类别是3， 请问可能是什么原因呢

mobilenetv2_tsm

dengfenglai321 commented 2 years ago

我尝试换了backbone和数据标签编号， 模型依然是固定输出类别3， 即第三个类别分数最高。。。。

dengfenglai321 commented 2 years ago

是忘记加上model.eval()的原因。。。。。。。。。。。