使用pytorch时遇到下面的问题
RuntimeError: split_size can only be 0 if dimension size is 0, but got dimension size of 2
原因:
训练的batch size 比使用的GPU数量少,导致上述问题。
解决办法
增加batch size数值,保证为GPU数量整数倍。
参考:
1.https://discuss.pytorch.org/t/concatenating-images/40961/10