Traceback (most recent call last):
File “train.py”, line 58, in
model.optimize_parameters()
File “/content/gdrive/My Drive/SNE-RoadSeg-master/models/roadseg_model.py”, line 59, in optimize_parameters
self.forward()
File “/content/gdrive/My Drive/SNE-RoadSeg-master/models/roadseg_model.py”, line 50, in forward
self.output = self.netRoadSeg(self.rgb_image, self.another_image)
File “/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py”, line 493, in call
result = self.forward(*input, **kwargs)
File “/usr/local/lib/python3.7/dist-packages/torch/nn/parallel/data_parallel.py”, line 150, in forward
return self.module(*inputs[0], **kwargs[0])
File “/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py”, line 493, in call
result = self.forward(*input, **kwargs)
File “/content/gdrive/My Drive/SNE-RoadSeg-master/models/networks.py”, line 236, in forward
x3_2 = self.conv3_2(torch.cat([x3_0, x3_1, self.up4_1(x4_1)], dim=1))
File “/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py”, line 493, in call
result = self.forward(*input, **kwargs)
File “/content/gdrive/My Drive/SNE-RoadSeg-master/models/networks.py”, line 94, in forward
x = self.up(x)
File “/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py”, line 493, in call
result = self.forward(*input, **kwargs)
File “/usr/local/lib/python3.7/dist-packages/torch/nn/modules/upsampling.py”, line 131, in forward
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
File “/usr/local/lib/python3.7/dist-packages/torch/nn/functional.py”, line 2563, in interpolate
return torch._C._nn.upsample_bilinear2d(input, _output_size(2), align_corners)
RuntimeError: CUDA out of memory. Tried to allocate 60.00 MiB (GPU 0; 11.17 GiB total capacity; 10.34 GiB already allocated; 30.81 MiB free; 489.75 MiB cached)
Your GPU is out of memory. You can decrease batch size to save some.