batchsize = 1 传入模型的问题

[britney-code]

作者，你好 ，batchsize = 1的时候，传入模型获得logits的有的模型维度为[1,1001] 有的模型维度为[1001] 如何修改呢？

[ylhz]

你好，可以在torch_net中对应的模型def forward()中调整为统一的输出格式。也可以在输出后调整处理logits的代码。

[zchoi]

solid