meraks / Swin-Transformer-1D

Swin-Transformer 1D implements
MIT License
38 stars 2 forks source link

你好还是关于输出格式的问题 #2

Open GUESSZERO4 opened 2 years ago

GUESSZERO4 commented 2 years ago

输入的Size(2,32,2000)在模型输出后的Size为什么还是(2,9)。它的第一个维度2代表着什么?还是batch吗。输出不应该是对于几个类别的预测概率吗?

meraks commented 2 years ago

输入size的含义为[batch, channel, seq_len],输出size的含义为[batch, num_classes],至于第一个维度2就是代表的batch size。