造成这种情况的原因主要是电脑的内存不足导致的。可以调整以下参数解决问题。
方法一:
结合实际,缩小batch-size或者workers的值。如把workers的值由8改为2或1或0进行尝试,也可以将batch-size的值由16改为8或4(修改batch-size的数值可能影响训练结果的精度等)
方法二:
修改电脑用于训练的内存:
右键“我的电脑”——高级系统设置——高级——设置
——高级——更改——选择对应的硬盘(训练所用的盘)——自定义大小(结合可用空间设置初始大小和最大值)
初始大小和最大值的设定结合自己的可用空间,可多次尝试。