Giters

mit-han-lab / temporal-shift-module

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding

Home Page:https://arxiv.org/abs/1811.08383

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

模型输出类别固定

dengfenglai321 opened this issue · comments

commented

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

mobilenetv2_tsm

commented

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

commented

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