torch yolov3 minibatch 失败

minibatch失败了，原因是内存溢出了，代码：

loss += model(sub_imgs, sub_targets)

# -*- coding:utf-8 -*-
from __future__ import division

from models import *
from utils.utils import *
from utils.datasets import *
from utils.parse_config 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/80782927