建议把网络中下x.view那行修改为x = x.view(x.shape[0], -1)。
Expected input batch_size (1) to match target batch_size (16).
最新推荐文章于 2023-08-23 18:31:06 发布
建议把网络中下x.view那行修改为x = x.view(x.shape[0], -1)。