Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

error while training #39

Open
dnomyaR23 opened this issue Oct 27, 2021 · 0 comments
Open

error while training #39

dnomyaR23 opened this issue Oct 27, 2021 · 0 comments

Comments

@dnomyaR23
Copy link

dnomyaR23 commented Oct 27, 2021

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)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant