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)
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)