This is an image during a debugging when the error occurs.
image490×749 30.2 KB
The size of the image when this error happens is: (8, 1, 96, 609)
In the expect statement an error occured when print(img) get called, which is:
    print(img)
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\tensor.py", line 179, in __repr__
    return torch._tensor_str._str(self)
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 372, in _str
    return _str_intern(self)
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 352, in _str_intern
    tensor_str = _tensor_str(self, indent)
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 241, in _tensor_str
    formatter = _Formatter(get_summarized_data(self) if summarize else self)
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 273, in get_summarized_data
    return torch.stack([get_summarized_data(x) for x in (start + end)])
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 273, in <listcomp>
    return torch.stack([get_summarized_data(x) for x in (start + end)])
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 275, in get_summarized_data
    return torch.stack([get_summarized_data(x) for x in self])
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 275, in <listcomp>
    return torch.stack([get_summarized_data(x) for x in self])
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 273, in get_summarized_data
    return torch.stack([get_summarized_data(x) for x in (start + end)])
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 273, in <listcomp>
    return torch.stack([get_summarized_data(x) for x in (start + end)])
  File "C:\Users\Mohammed\.conda\envs\dl_env\lib\site-packages\torch\_tensor_str.py", line 266, in get_summarized_data
    return torch.cat((self[:PRINT_OPTS.edgeitems], self[-PRINT_OPTS.edgeitems:]))
RuntimeError: CUDA error: unspecified launch failure