这个错误主要是因为,数据的数量不能整除batch size的数量,改正方法有两种
1. 将batch size改称可以被 数据量整除即可。
2. 将dataloader 中的 drop_last改称False也可。
推荐使用第二种方法更加简单。
这个错误主要是因为,数据的数量不能整除batch size的数量,改正方法有两种
1. 将batch size改称可以被 数据量整除即可。
2. 将dataloader 中的 drop_last改称False也可。
推荐使用第二种方法更加简单。