学习笔记
错误:训练时显存不足
原因:
1、改变输入图片的大小
input_shape = [320, 320]-----显存不足的话就调小一点(大小为32的倍数)
2、改变batch_size的大小
显存不足就调小-----最小为1(有的是最小为2)-------#不能为0#
3、改变num_workers(线程)的数值
选择是否打开多线程模式 或者 打开几个线程
错误:训练时显存不足
原因:
1、改变输入图片的大小
input_shape = [320, 320]-----显存不足的话就调小一点(大小为32的倍数)
2、改变batch_size的大小
显存不足就调小-----最小为1(有的是最小为2)-------#不能为0#
3、改变num_workers(线程)的数值
选择是否打开多线程模式 或者 打开几个线程