CUDA out of memory

[xuhao-anhe]

Traceback (most recent call last): File "train.py", line 438, in train(hyp, opt, device, tb_writer) File "train.py", line 252, in train pred = model(imgs) File "/home/yijie/anaconda3/envs/yolov4/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/media/yijie/code/xuhao/PyTorch_YOLOv4-master/models/models.py", line 293, in forward return self.forward_once(x) File "/media/yijie/code/xuhao/PyTorch_YOLOv4-master/models/models.py", line 347, in forward_once x = module(x) File "/home/yijie/anaconda3/envs/yolov4/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/home/yijie/anaconda3/envs/yolov4/lib/python3.8/site-packages/torch/nn/modules/container.py", line 117, in forward input = module(input) File "/home/yijie/anaconda3/envs/yolov4/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/home/yijie/anaconda3/envs/yolov4/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 419, in forward return self._conv_forward(input, self.weight) File "/home/yijie/anaconda3/envs/yolov4/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 415, in _conv_forward return F.conv2d(input, weight, self.bias, self.stride, RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 7.93 GiB total capacity; 3.25 GiB already allocated; 10.44 MiB free; 6.90 GiB reserved in total by PyTorch)

[abcc235]

调小banch