先后尝试 .contiguous(),num_workers,最后发现是有个数据集输入特征维度不对。

更多推荐