- 博客(1)
- 收藏
- 关注
原创 解决RuntimeError: stack expects a non-empty Tensorlist问题
在batch_size时,可能某些地方需要batch>1的时候才能运行,但在创建dataloader时,可能最后一个batch的大小正好为1,所以会导致这种错误。为了解决这种情况,将dataloader的参数drop_last设置为True就可以了。
2022-12-27 14:05:46
5614
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人