Closed chetanmreddy closed 3 years ago
Usually I do not get such errors, however cuda errors are somewhat cryptical in most cases. Did you change anything in the code with respect to the standard configuration?
Also: Did you already try to run the code on a CPU, just to verify that it is running? If you get an error on the CPU sometimes it is more readable than on the GPU. The code automatically detects if it can be run on GPU and if not available it runs on the CPU.
I did not change anything in the code except for correct paths to data.
I tried running on CPU and it trains without any errors. For now, I will keep this issue open and try to debug around. Thank You.
After following some solutions on PyTorch discussions forum, the error is changed to this:
Traceback (most recent call last):
File "train.py", line 372, in
UPDATE: It seems like the error is happening in the segmentation part. Can you please elaborate on how to prepare cityscapes dataset?
Any idea why this might be happening?
`Starting initialization Downloading: "https://download.pytorch.org/models/resnet18-5c106cde.pth" to /root/.cache/torch/checkpoints/resnet18-5c106cde.pth Loading training dataset metadata: